tensorflow - 使用 tf.app.run() 从类中调用 main 函数

Question

问题很简单，我正在尝试使用 tf.app.run() 来调用类中的 main 函数。但是，下面的代码给了我一个错误。如有任何帮助，我们将不胜感激。

将 tensorflow 导入为 tf 导入系统

# Where to find data
tf.app.flags.DEFINE_string('f1', '', 'feature 1')
tf.app.flags.DEFINE_string('f2', '', 'feature 2')

FLAGS = tf.app.flags.FLAGS

class Test(object):
    def __init__(self):
        pass
    def main(self, args):
        print(FLAGS.__flag.iteritems())

def main(args):
    test = Test()
    test.main(args)

if __name__ == '__main__':
    tf.app.run(main)

这是错误:

Traceback (most recent call last):
  File "test.py", line 21, in <module>
    tf.app.run(main)
  File "/Users/yaserkeneshloo/anaconda/envs/env27/lib/python2.7/site-packages/tensorflow/python/platform/app.py", line 126, in run
    _sys.exit(main(argv))
  File "test.py", line 18, in main
    test.main(args)
  File "test.py", line 14, in main
    print(FLAGS.__flag.iteritems())
  File "/Users/yaserkeneshloo/anaconda/envs/env27/lib/python2.7/site-packages/tensorflow/python/platform/flags.py", line 85, in __getattr__
    return wrapped.__getattr__(name)
  File "/Users/yaserkeneshloo/anaconda/envs/env27/lib/python2.7/site-packages/absl/flags/_flagvalues.py", line 470, in __getattr__
    raise AttributeError(name)
AttributeError: _Test__flag

Answer 1

问题是，当您尝试从类内部访问以双下划线开头的 FLAGS 属性时，它会在该属性前面加上类名前缀。因此，它尝试查找不存在的 FLAGS._Test__flags，而不是 FLAGS.__flags。

对于设置或获取值都是如此。因此，如果您从类外部设置值，则必须在其前面添加 _Test 前缀(因为您将类命名为 Test。)。如果您在类内部设置标志，您不需要添加前缀，因为它在分配值时也会自动添加前缀。

所以基本上你的代码没有任何问题，因为你自己的标志不以双下划线开头，除非你不能使用内部 __flags 属性来打印所有标志。不过，您可以单独访问它们。

请参阅下面的代码以获取详细示例。 (此外，DEFINE 行中的默认值是第二个参数，而不是第三个。)

# Where to find data
tf.app.flags.DEFINE_string('_Test__f1', 'feature 1', 'feature 1')
tf.app.flags.DEFINE_string('__f2', 'feature 2', 'feature 2')

FLAGS = tf.app.flags.FLAGS    

print( FLAGS.__f2 ) # prints "feature 2"

class Test(object):
    def __init__(self):
        pass
    def main(self, args):
        print( FLAGS.__f1 ) # prints "feature 1"
        FLAGS.__f1 = 'foobar' # assignment works the same way
        print( FLAGS.__f1 ) # prints "foobar"
        print( FLAGS.__f2 ) # AttributeError: _Test__f2

def main(args):
    test = Test()
    test.main(args)

if __name__ == '__main__':
    tf.app.run(main)