前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Tensorflow教程(十四) 命令行参数tf.flags的使用

Tensorflow教程(十四) 命令行参数tf.flags的使用

作者头像
致Great
发布2018-10-12 09:54:11
2K0
发布2018-10-12 09:54:11
举报
文章被收录于专栏:程序生活

实例1

代码语言:javascript
复制
import tensorflow as tf

flags = tf.flags
FLAGS=flags.FLAGS
flags.DEFINE_integer('data_num', 100, """Flag of type integer""")
flags.DEFINE_string('img_path', './img', """Flag of type string""")
FLAGS.flag_values_dict()

def main():
    print(FLAGS.data_num, FLAGS.img_path)


if __name__ == '__main__':
    tf.app.run()

在命令行中执行:python cnn_model.py

实例2

代码语言:javascript
复制
import tensorflow as tf

FLAGS = tf.app.flags.FLAGS
tf.app.flags.DEFINE_integer('data_num', 100, """Flag of type integer""")
tf.app.flags.DEFINE_string('img_path', './img', """Flag of type string""")


def main(argv):
    print(FLAGS.data_num, FLAGS.img_path)


if __name__ == '__main__':
    tf.app.run()

这个和实例1中的代码几乎一样,不同的地方是:

  • 实例1:flags = tf.flags
  • 实例2:flags = tf.app.flags

但是实例2中的但是在pycharm提示如下警告:

自己查了很多关于tensorflow命令行教程,大多都和实例2相似,对于强迫症的我,忍不了,以后统一成实例1用了。

参考资料

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018.10.11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 实例1
  • 实例2
  • 参考资料
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档