Pycharm 本地运行时设置flags参数

举报
技术火炬手 发表于 2020/06/28 16:19:34 2020/06/28
【摘要】 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】进入设置

image.png

进入设置后,选择你需要配置flags参数的py文件,然后再点击Parameters框右侧的图标,添加flag

image.png

按照下图中的格式,在框内输入你的参数

image.png

上面是我的参数,可以供大家参考。

image.png

配置好参数后,点击“OK”确认即可。最后,点击鼠标右键,选择【Run ‘run_classifier’ with Coverage】就可以使用刚刚配置的参数运行Python文件了

image.png

运行Python文件,可以方便地进行debug,达到查看模型内部详细信息的目的。

需要注意的是,这只是方便进行本地调试的方法,如果代码需要到服务器运行,还是需要使用sh脚本进行部署的

---------------------------------

本文转自“机器在学习”

原文链接:http://www.siyuanblog.com/?p=115342

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。