Tensorflow教程(十四) 命令行参数tf.flags的使用
实例1
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
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 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
