File size: 1,624 Bytes
ab6c03c | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | #!/bin/bash
export MODEL_PATH=/gluster/zhihan/backup/dna/690/6
for model in $(ls $MODEL_PATH)
do
export MODEL="$model"
export CHECKPOINT=$(ls $MODEL_PATH/$MODEL | head -1)
if [ ! -d "/gluster/zhihan/DNABERT/examples/data/ori_results/$MODEL" ]
then
python run_finetune.py \
--model_type dna \
--tokenizer_name=dna6 \
--model_name_or_path $MODEL_PATH/$MODEL/$CHECKPOINT \
--task_name dnaprom \
--do_predict \
--data_dir /gluster/zhihan/DNABERT/examples/data/ori \
--max_seq_length 110 \
--per_gpu_pred_batch_size=256 \
--output_dir $MODEL_PATH/$MODEL/$CHECKPOINT \
--predict_dir /gluster/zhihan/DNABERT/examples/data/ori_results/$MODEL \
--fp16 \
--n_process 96
fi
done
for model in $(ls $MODEL_PATH)
do
export MODEL="$model"
export CHECKPOINT=$(ls $MODEL_PATH/$MODEL | head -1)
if [ ! -d "/gluster/zhihan/DNABERT/examples/data/mut_results/$MODEL" ]
then
python run_finetune.py \
--model_type dna \
--tokenizer_name=dna6 \
--model_name_or_path $MODEL_PATH/$MODEL/$CHECKPOINT \
--task_name dnaprom \
--do_predict \
--data_dir /gluster/zhihan/DNABERT/examples/data/mut \
--max_seq_length 110 \
--per_gpu_pred_batch_size=256 \
--output_dir $MODEL_PATH/$MODEL/$CHECKPOINT \
--predict_dir /gluster/zhihan/DNABERT/examples/data/mut_results/$MODEL \
--fp16 \
--n_process 96
fi
done
|