File size: 759 Bytes
65bd8af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
export CUDA_VISIBLE_DEVICES=0
export WANDB_API_KEY=b7b6ecceb6854bd12f58809f18264f979509d13b
python predict.py \
-lr 1e-4 \
-batch_size 16 \
-d_node 1288 \
-margin 10 \
-grad_clip 2 \
-max_epochs 10 \
-num_heads 8 \
-dropout 0.2 \
-binder 'AQTVPYGIPLIKADKVQAQGFKGANVKVAVLDTGIQASHPDLNVVGGASFVAGEAYNTDGNGHGTHVAGTVAALDNTTGVLGVAPSVSLYAVKVLNSSGSGSYSGIVSGIEWATTNGMDVINMSLGGASGSTAMKQAVDNAYARGVVVVAAAGNSGNSGSTNTIGYPAKYDSVIAVGAVDSNSNRASFSSVGAELEVMAPGAGVYSTYPTNTYATLNGTSMASPHVAGAAALILSKHPNLSASQVRNRLSSTATYLGSSFYYGKGLINVEAAAQ' \
-wildtype 'TEFGSELKSFPEVVGKTVDQAREYFTLHYPQYNVYFLPEGSPVTLDLRYNRVRVFYNPGTNVVNHVPHVG' \
-mutant 'TEFGSELKSFPEVVGKTVDQAREYFTLHYPQYNVYFLPEGSPVTSDLRYNRVRVFYNPGTNVVNHVPHVG' \
-sm '/home/tc415/muPPIt_embedding/checkpoints/improved_train_0/acc=0.6'
|