0
点赞
收藏
分享

微信扫一扫

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用了。

参考资料

  • cnn-re-tf/train.py at master · may-/cnn-re-tf
  • <code>FLAGS</code> in Python & Tensorflow
  • Command line flags: Python and Tensorflow – Zoe Hartman – Medium
  • TensorFlow中的小知识:tf.flags.DEFINE_xxx() - spring_willow的博客 -
举报

相关推荐

0 条评论