BiBERTa / config /config_hparam.json
jinysun's picture
Upload 2 files
3ccfb9b
raw
history blame
533 Bytes
{ "name": "biomarker_log",
"d_model_name" : "DeepChem/ChemBERTa-10M-MTR",
"p_model_name" : "DeepChem/ChemBERTa-77M-MLM",
"gpu_ids" : "0",
"model_mode" : "train",
"load_checkpoint" : "./checkpoint/bindingDB/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" : [512, 128, 64, 1],
"dropout" : 0.1,
"loss_fn" : "MSE",
"traindata_rate" : 1.0,
"pretrained": {"chem":true, "prot":true},
"num_seed" : 111
}