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