flags的解析方法

最新推荐文章于 2020-12-25 20:52:45 发布

ei1994

最新推荐文章于 2020-12-25 20:52:45 发布

阅读量3.8k

点赞数 1

分类专栏： python

本文链接：https://blog.csdn.net/ei1990/article/details/76423277

版权

python 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

1、使用argparse包

import argparse
parser = argparse.ArgumentParser()
parser.add_argument(
      '--learning_rate',
      type=float,
      default=0.01,
      help='Initial learning rate.'
  )
parser.add_argument(
      '--max_steps',
      type=int,
      default=2000,
      help='Number of steps to run trainer.'
  )
FLAGS, unparsed = parser.parse_known_args()
# 输出时，
print（FLAGS.learning_rate)  #输出=0.01

2、使用tf的包（1）

import tensorflow as tf
FLAGS = tf.app.flags.FLAGS

tf.app.flags.DEFINE_string('train_dir', 'tmp/train/',
                           """Directory where to write event logs """
                           """and checkpoint.""")
tf.app.flags.DEFINE_integer('max_steps', 700,
                            """Number of batches to run.""")
tf.app.flags.DEFINE_boolean('log_device_placement', False,
                            """Whether to log device placement.""")

print(FLAGS.max_steps)   # 输出=700

3、使用tf的包（2）——简洁用法

from tensorflow import flags
flags.DEFINE_float("threshold", 0.1, "detection threshold")
flags.DEFINE_string("model", "", "configuration of choice")
flags.DEFINE_string("trainer", "rmsprop", "training algorithm")
FLAGS = flags.FLAGS

print(FLAGS.threshold)   # 输出=0.1