muPPIt / scripts /predict.sh
AlienChen's picture
Upload 139 files
65bd8af verified
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'