Spaces:
Runtime error
Runtime error
File size: 941 Bytes
8918ac7 |
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 |
### Dataset
# ESMFold & AlphaFold2: DeepLocBinary DeepLocMulti MetalIonBinding EC Thermostability
# ESMFold: DeepSol DeepSoluE
# No structure: FLIP_AAV FLIP_GB1
### Protein Language Model (PLM)
# facebook: esm2_t30_150M_UR50D esm2_t33_650M_UR50D esm2_t36_3B_UR50D
# RostLab: prot_bert prot_bert_bfd prot_t5_xl_uniref50 prot_t5_xl_bfd ankh-base ankh-large
export HF_ENDPOINT=https://hf-mirror.com
dataset=DeepLocBinary
pdb_type=AlphaFold2
plm_source=facebook
plm_model=esm2_t30_150M_UR50D
lr=5e-4
python src/train.py \
--plm_model $plm_source/$plm_model \
--dataset_config data/$dataset/"$dataset"_"$pdb_type"_HF.json \
--learning_rate $lr \
--num_epochs 50 \
--batch_token 12000 \
--gradient_accumulation_steps 8 \
--patience 3 \
--structure_seq foldseek_seq,ss8_seq \
--output_dir debug/$dataset/$plm_model \
--output_model_name ses-adapter_"$pdb_type"_lr"$lr"_bt12k_ga8.pt
|