TensorFlow中的tf.app.flags使用

最新推荐文章于 2020-10-15 14:21:18 发布

醉小义

最新推荐文章于 2020-10-15 14:21:18 发布

阅读量941

点赞数

分类专栏： tensorflow 文章标签： tf.app.flags

本文链接：https://blog.csdn.net/qq_30638831/article/details/80789654

版权

tensorflow 专栏收录该内容

53 篇文章 0 订阅

订阅专栏

import tensorflow as tf
#参数说明  (变量名，值，说明)
tf.app.flags.DEFINE_string(
    'master', 'variable', 'The address of the TensorFlow master to use.')

tf.app.flags.DEFINE_integer(
    'worker_replicas', 1, 'Number of worker replicas.')

tf.app.flags.DEFINE_float(
    'weight_decay', 0.00004, 'The weight decay on the model weights.')

tf.app.flags.DEFINE_boolean(
    'clone_on_cpu', False,'Use CPUs to deploy clones.')

FLAGS = tf.app.flags.FLAGS

print(FLAGS.master)
print(FLAGS.worker_replicas)
print(FLAGS.weight_decay)
print(FLAGS.clone_on_cpu)

'''
  输出结果:
        variable
        1
        4e-05
        False
'''


'''
    访问方式1：
        需要改变的参数值的需要用--变量名=value
        python3 flags.py --master=123  --worker_replicas=3
    访问方式2：
        直接运行,进行操作

'''