Tensorflow学习之tf.app.flags

最新推荐文章于 2023-04-01 09:16:45 发布

未生无

最新推荐文章于 2023-04-01 09:16:45 发布

阅读量200

点赞数

文章标签： tensorflow

原文链接：https://www.jianshu.com/p/55cbd3753ee8

版权

tf.app.flags主要用于处理命令行参数的传递工作

通过tf.app.flags来调用这个flags.py文件，这样我们就可以用flags.DEFINE_interger/float()来添加命令行参数，而FLAGS=flags.FLAGS可以实例化这个解析参数的类从对应的命令行参数取出参数。

import tensorflow as tf  

flags = tf.app.flags
flags.DEFINE_string('data_dir', '/tmp/mnist', 'Directory with the MNIST data.')
flags.DEFINE_integer('batch_size', 5, 'Batch size.')
flags.DEFINE_integer('num_evals', 1000, 'Number of batches to evaluate.')
FLAGS = flags.FLAGS

print(FLAGS.data_dir, FLAGS.batch_size, FLAGS.num_evals)

在命令行中输入test.py -h就可以查看帮助信息，也就是Directory with the MNIST data.，Batch size和Number of batches to evaluate这样的消息。
在命令行中输入test.py --batchsize 10就可以将batch_size的值修改为10！