Pycharm 本地运行时设置flags参数
【摘要】 Python编程中flags可以帮助我们通过命令行来动态的更改代码中的参数。
Python编程中flags可以帮助我们通过命令行来动态的更改代码中的参数。以bert为例,在运行代码的时候,有很多flag需要配置
export BERT_BASE_DIR=/Users/mtdp/Desktop/code/data/tmp/chinese_L-12_H-768_A-12
export SERVE_DIR=/Users/mtdp/Desktop/code/data/tmp/bert_multi_data/predict
python main.py \
--task_name=fserve \
--do_train=True \
--do_eval=False \
--do_predict=True \
--do_export=True \
--data_dir=$SERVE_DIR \
--vocab_file=$BERT_BASE_DIR/vocab.txt \
--bert_config_file=$BERT_BASE_DIR/bert_config.json \
--init_checkpoint= \
--max_seq_length=300 \
--train_batch_size=1 \
--learning_rate=5e-5 \
--num_train_epochs=1 \
--output_dir=/Users/lisiyuan/Desktop/code/data/tmp/serve_output/
通常来说,我们会把上面的这些代码写到一个sh脚本中,通过执行sh脚本,就可以启动bert的训练过程,但在代码开发过程中,我们经常会需要进行debug,查看代码中一些变量的值,shape等属性,这对于熟悉模型结构而言,非常重要,使用sh脚本启动训练过程,不是很方便进行debug。
我注意到Eclipse中,可以设置flags,但平时我主要还是使用Pycharm进行开发,找了一下,发现「tensorflow flags 最全 pycharm 运行时输入参数 list string int float」有提到使用pycharm配置flags,但说的不是特别详细,因此本文在上述博文的基础上进行了补充说明。
首先,点击【Run】–>【Edit Configurations】进入设置
进入设置后,选择你需要配置flags参数的py文件,然后再点击Parameters框右侧的图标,添加flag
按照下图中的格式,在框内输入你的参数
上面是我的参数,可以供大家参考。
配置好参数后,点击“OK”确认即可。最后,点击鼠标右键,选择【Run ‘run_classifier’ with Coverage】就可以使用刚刚配置的参数运行Python文件了
运行Python文件,可以方便地进行debug,达到查看模型内部详细信息的目的。
需要注意的是,这只是方便进行本地调试的方法,如果代码需要到服务器运行,还是需要使用sh脚本进行部署的
---------------------------------
本文转自“机器在学习”
【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
作者其他文章
评论(0)