旗帜在张量流中意味着什么?

时间:2017-10-15 16:30:05

标签: tensorflow

我是tensorflow的新手,我已从其网站上阅读了一些代码。我想知道代码中FLAGS的含义是什么?

例如,在fully_connected_feed.py的第78行,输入为FLAGS.batch_sizeFLAGS.fake_data。然后我在FLAGS的定义中读到了这个:

# Basic model parameters as external flags.
FLAGS = None

我对这个变量感到困惑。

1 个答案:

答案 0 :(得分:2)

用于存储命令行参数。查看这些行:

parser = argparse.ArgumentParser()
parser.add_argument(
      '--learning_rate',
      type=float,
      default=0.01,
      help='Initial learning rate.'
  )
... 
FLAGS, unparsed = parser.parse_known_args()

您还可以看到它允许您在命令行上定义默认值并创建帮助说明,以便其他人知道哪些参数接受您的cli而无需查看代码。