Create config.yaml

config.yaml

@@ -0,0 +1,13 @@
+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"