pCoMole/gfp/classifier_ckpt/config.json

{
  "csv_path": "/scratch/pranamlab/tong/data/gfp/classifier_total_data.csv",
  "out_dir": "/scratch/pranamlab/tong/data/gfp/esm_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"
}