hyperparameters: | |
lr: 1e-5 | |
weight_decay: 1e-5 | |
dropout: 0.1 | |
epochs: 30 | |
batch_size: 16 | |
seed: 5471 | |
logging: | |
model_type: "baseline" | |
model_version: "test_lr=${hyperparameters.lr}_batch_size=${hyperparameters.batch_size}_dropout=${hyperparameters.dropout}_wd=${hyperparameters.weight_decay}_seed=${hyperparameters.seed}" | |
model_save_path: "./models" | |
accelerate_save_path: "./output" | |
tensorboard_logdir: "./log" |