在python

时间:2019-07-06 19:26:28

标签: python tensorflow

我想在Jupyter Notebook中而不是从终端运行TensorFlow Object Detection API。特别是,我想通过调用具有train.py功能的main来开始培训

def main(_):
   pass

在使用python train.py <additional flags>运行文件时被调用,原因是文件中的这一部分:

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

我发现它在从终端中传递的标志中设置了全局变量后,调用了main函数以开始运行:

flags = tf.app.flags
flags.DEFINE_string('master', '', 'Name of the TensorFlow master to use.')
...

有没有一种方法可以设置全局变量flags,而无需更改脚本train.py,然后通过导入文件来调用main函数。不幸的是,我在tf.app.flags上找不到任何文档。

0 个答案:

没有答案