src/configs/train.yml

plm: esm2_t33_650M_UR50D
train_csv: data/train_addition.csv
test_csv: data/test.csv
batch_size: 32
epochs: 100
lr: 0.001
wd: 0
num_workers: 4
amp: true
seed: 42
nfolds: 5
ckpt_dir: checkpoints
patience: 20