欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

Python随记:如何在pycham下调试需要指定参数的脚本

时间:2023-05-20
以P-Tuning V2的run_rte_roberta.sh为例:

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",插入要指定的参数:


Step 4、参数添加完成后,点击“OK”确认返回 Step 5、右键点击run.py的"Debug run"

这样我们就能正常debug调试代码了

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。