File size: 845 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 | export MODEL_PATH=/home/zhihan/6
# for cp in $(ls $MODEL_PATH)
# do
# cd $MODEL_PATH/$cp
# mv checkpoin* checkpoint-0
# done
for model in $(ls $MODEL_PATH | head -345)
do
export MODEL="$model"
export CHECKPOINT=$(ls $MODEL_PATH/$MODEL)
CUDA_VISIBLE_DEVICES=0 python run_finetune.py \
--model_type dna \
--tokenizer_name=dna6 \
--model_name_or_path $MODEL_PATH/$MODEL/$CHECKPOINT \
--task_name dnaprom \
--do_visualize \
--visualize_data_dir /home/zhihan/data/uce/processed/ \
--visualize_models 6 \
--data_dir /home/zhihan/data/uce/processed/ \
--max_seq_length 110 \
--per_gpu_pred_batch_size=16 \
--output_dir $MODEL_PATH/$MODEL/$CHECKPOINT \
--predict_dir /home/zhihan/data/uce/results/$MODEL \
--n_process 24
done |