我正在尝试测试一个简短的程序,该程序非常简单,如下所示
import numpy as np
import tensorflow as tf
flags = tf.app.flags
FLAGS = flags.FLAGS
import tensorvision.train as train
import tensorvision.utils as utils
flags.DEFINE_string('name', None,
'Append a name Tag to run.')
flags.DEFINE_string('hypes', 'hypes/medseg.json',
'File storing model parameters.')
if __name__ == '__main__':
tf.app.run()
但是,运行该程序会出现以下错误消息,
Traceback (most recent call last):
File "train.py", line 43, in <module>
tf.app.run()
File "/devl/tensorflow/tf_0.12/lib/python3.4/site- packages/tensorflow/python/platform/app.py", line 39, in run
main = main or sys.modules['__main__'].main
AttributeError: 'module' object has no attribute 'main'
您的文件中需要一个“def main(args)”函数:
import numpy as np
import tensorflow as tf
flags = tf.app.flags
FLAGS = flags.FLAGS
import tensorvision.train as train
import tensorvision.utils as utils
def main(args):
flags.DEFINE_string('name', None,
'Append a name Tag to run.')
flags.DEFINE_string('hypes', 'hypes/medseg.json',
'File storing model parameters.')
if __name__ == '__main__':
tf.app.run()
或者 tf.app.run() 可以调用外部函数
tf.app.run(my_func)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)