| { | |
| "csv_path": "", | |
| "out_dir": "gfp/classifier_ckpt", | |
| "seed": 42, | |
| "esm_name": "facebook/esm2_t33_650M_UR50D", | |
| "batch_size": 8, | |
| "num_workers": 4, | |
| "max_length": 1024, | |
| "epochs": 20, | |
| "lr": 0.0002, | |
| "weight_decay": 0.01, | |
| "warmup_ratio": 0.1, | |
| "min_lr_ratio": 0.1, | |
| "grad_clip": 1.0, | |
| "mlp_hidden": 512, | |
| "mlp_layers": 2, | |
| "dropout": 0.2, | |
| "val_ratio": 0.15, | |
| "test_ratio": 0.15, | |
| "amp": false, | |
| "metric_for_best": "auroc" | |
| } |