demo/training_args.json

{
  "plm_model": "facebook/esm2_t33_650M_UR50D",
  "training_method": "freeze",
  "pooling_method": "mean",
  "learning_rate": 0.0005,
  "num_epochs": 5,
  "max_seq_len": -1,
  "gradient_accumulation_steps": 1,
  "warmup_steps": 0,
  "scheduler": null,
  "output_model_name": "demo.pt",
  "output_dir": "demo",
  "patience": 10,
  "num_workers": 4,
  "max_grad_norm": -1,
  "dataset_config": "data/FLIP_AAV/FLIP_AAV_des-mut_HF.json",
  "batch_token": 4000
}