AptaBLE / config.yaml
AtomBio's picture
Create config.yaml
24c4c83 verified
raw
history blame
424 Bytes
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"