export TASK_NAME=superglue export DATASET_NAME=rte export CUDA_VISIBLE_DEVICES=3 bs=64 lr=5e-3 dropout=0.1 psl=128 epoch=200 python3 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 --prefixStep 1. 在run.py中指定GPU使用的卡号:
os.environ['CUDA_VISIBLE_DEVICES'] = "3"Step 2.在run.py脚本上点击鼠标右键,依次选择"More Run/Debug"——>“Modify Run Configuration” Step 3. 编辑"Parameters",插入要指定的参数:
这样我们就能正常debug调试代码了



