Flexpert-Design/src/models/configs/ANMAwareFlexibilityProtTrans.yaml

checkpoint_path: /scratch/project/fta-24-31/koubapet/ProteinInvBench/flex_model_weights/conv_adaptor #finetuned_model_BS8.pth
data_jsonl_name: /NEWcath_ANM_gt_flex_annotated.jsonl #/debug_cath_ANM_gt_flex_annotated.jsonl #/cath_ANM_gt_flex_annotated.jsonl
half_precision: False #mixed_precision
gumbel_temperature: 0.2
num_labels: 1
add_pearson_loss: False
add_sse_loss: False
adaptor_architecture: 'conv'
enm_embed_dim: 128
enm_att_heads: 8
num_layers: 3
kernel_size: 5