tensorflow tf.app.flags.FLAGS 从命令行添加参数

最新推荐文章于 2022-08-02 20:43:31 发布

bboysky45

最新推荐文章于 2022-08-02 20:43:31 发布

阅读量774

点赞数

分类专栏： tensorflow 文章标签： tensorflo

tensorflow 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了如何使用TensorFlow的tf.app.flags.DEFINE_xxx()函数从命令行接收参数，通过定义不同类型如整数和字符串的标志(flag)，并在Python脚本中访问这些参数的值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用语句tf.app.flags.DEFINE_xxx()从命令行加入参数进去

举个例子

import tensorflow as tf
FLAGS = tf.app.flags.FLAGS
##下面是模型的参数
#定义整数型flag
tf.app.flags.DEFINE_integer(flag_name = 'batch_size',default_value = 128,
                            docstring='Path to the CIFAR-10 data directory.')

# 定义字符型flag
tf.app.flags.DEFINE_string(flag_name='data_dir', default_value='cifar-10-binary',
                           docstring='Path to the CIFAR-10 data directory.')

print(FLAGS.batch_size)
print(FLAGS.data_dir)

运行结果如下：