export TASK_NAME=superglueexport DATASET_NAME=rteexport CUDA_VISIBLE_DEVICES=3bs=64lr=5e-3dropout=0.1psl=128epoch=200python3 run.py --model_name_or_path roberta-large --task_name $TASK_NAME --dataset_name $DATASET_NAME --do_train --do_eval --max_seq_length 128 --per_device_train_batch_size $bs --learning_rate $lr --num_train_epochs $epoch --pre_seq_len $psl --output_dir checkpoints/$DATASET_NAME-roberta/ --overwrite_output_dir --hidden_dropout_prob $dropout --seed 11 --save_strategy no --evaluation_strategy epoch --prefix
Step 1、在run.py中指定GPU使用的卡号:os.environ['CUDA_VISIBLE_DEVICES'] = "3"
Step 2.在run.py脚本上点击鼠标右键,依次选择"More Run/Debug"——>“Modify Run Configuration” Step 3、编辑"Parameters",插入要指定的参数:
这样我们就能正常debug调试代码了