TensorFlow定义通用入口点脚本

2018-09-07 13:54 更新

#版权所有2015 TensorFlow作者.版权所有.

#根据Apache许可证版本2.0(“许可证”)许可;

#除非符合许可证,否则您不得使用此文件.

#您可以获得许可证的副本

#http      ://www.apache.org/licenses/LICENSE-2.0

#除非适用法律要求或书面同意软件

根据许可证分发的#分发在“按原样”基础上,

#无明示或暗示的任何种类的保证或条件.

#查看有关权限的特定语言的许可证

#许可证下的限制.

# =============================================== =============================

""通用入口点脚本.""

from __future__ import absolute_import

from __future__ import division

from __future__ import print_function

import sys as _sys

from tensorflow.python.platform import flags

from tensorflow.python.util.all_util import remove_undocumented

def _benchmark_tests_can_log_memory():

  return True

def run(main=None, argv=None):

  """Runs the program with an optional 'main' function and 'argv' list."""

  f = flags.FLAGS

  # Extract the args from the optional `argv` list.

  args = argv[1:] if argv else None

  # Parse the known flags from that list, or from the command

  # line otherwise.

  # pylint: disable=protected-access

  flags_passthrough = f._parse_flags(args=args)

  # pylint: enable=protected-access

  main = main or _sys.modules['__main__'].main

  # Call the main function, passing through any arguments

  # to the final program.

  _sys.exit(main(_sys.argv[:1] + flags_passthrough))

_allowed_symbols = [

    'run',

    # Allowed submodule.

    'flags',

]

remove_undocumented(__name__, _allowed_symbols)

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号