File size: 516 Bytes
3ccfb9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
{ "name": "biomarker_log",
"d_model_name" : "DeepChem/ChemBERTa-10M-MLM",
"p_model_name" : "DeepChem/ChemBERTa-10M-MTR",
"gpu_ids" : "0",
"model_mode" : "test",
"load_checkpoint" : "./OSC/test.ckpt",
"prot_maxlength" : 360,
"layer_limit" : true,
"max_epoch": 16,
"batch_size": 40,
"num_workers": 0,
"task_name" : "OSC",
"lr": 1e-4,
"layer_features" : [128, 128, 128, 1],
"dropout" : 0.1,
"loss_fn" : "MSE",
"traindata_rate" : 1.0,
"pretrained": {"chem":true, "prot":true},
"num_seed" : 111
}
|