如何使tf.app.run脚本成为可导入模块

时间:2019-08-16 01:40:10

标签: python tensorflow

tensorflow脚本通常具有以下结构。 如何将脚本转换成可导入的模块,以便可以从另一个python脚本调用主函数?

flags = tf.app.flags
flags.DEFINE_string('master', '', 'Name of the TensorFlow master to use.')
flags.DEFINE_integer('task', 0, 'task id')
flags.DEFINE_integer('num_clones', 1, 'Number of clones to deploy per worker.')
....

def main(_):
   ...

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

0 个答案:

没有答案