示例#1
0
def run_main():
    """Initializes flags and calls main()."""
    program.setup_environment()

    if getattr(tf, '__version__', 'stub') == 'stub':
        print(
            "TensorFlow installation not found - running with reduced feature set.",
            file=sys.stderr)

    tensorboard = program.TensorBoard(
        default.get_plugins() + default.get_dynamic_plugins(),
        program.get_default_assets_zip_provider(),
        subcommands=[uploader_main.UploaderSubcommand()])
    try:
        from absl import app
        # Import this to check that app.run() will accept the flags_parser argument.
        from absl.flags import argparse_flags
        app.run(tensorboard.main, flags_parser=tensorboard.configure)
        raise AssertionError("absl.app.run() shouldn't return")
    except ImportError:
        pass
    except base_plugin.FlagsError as e:
        print("Error: %s" % e, file=sys.stderr)
        sys.exit(1)

    tensorboard.configure(sys.argv)
    sys.exit(tensorboard.main())
示例#2
0
def run_main():
    """Initializes flags and calls main()."""
    program.setup_environment()

    if getattr(tf, '__version__', 'stub') == 'stub':
        # Unless the user has explicitly requested running without TensorFlow by setting the
        # TENSORBOARD_NO_TF environment variable, we check for TensorFlow here so that if it's
        # missing we generate a clear and immediate error rather than partial functionality.
        # TODO(#2027): Remove environment check once we have placeholder UI
        if os.getenv('TENSORBOARD_NO_TF') is None:
            import tensorflow  # pylint: disable=unused-import
        logger.warn(
            "TensorFlow installation not found - running with reduced feature set."
        )

    tensorboard = program.TensorBoard(
        default.get_plugins(), program.get_default_assets_zip_provider())
    try:
        from absl import app
        # Import this to check that app.run() will accept the flags_parser argument.
        from absl.flags import argparse_flags
        app.run(tensorboard.main, flags_parser=tensorboard.configure)
        raise AssertionError("absl.app.run() shouldn't return")
    except ImportError:
        pass
    except base_plugin.FlagsError as e:
        print("Error: %s" % e, file=sys.stderr)
        sys.exit(1)

    tensorboard.configure(sys.argv)
    sys.exit(tensorboard.main())
示例#3
0
def _wsgi_app_v2(
    TensorBoardWSGIApp,
    logdir,
    plugins,
    reload_interval=None,
    path_prefix="",
    tensorboard_options=None,
):
    """WSGI app for tensorboard>=2.3.0."""
    with warnings.catch_warnings():
        warnings.simplefilter("ignore", Warning)
        # pylint: disable=no-name-in-module
        from tensorboard.backend.event_processing import data_ingester
        from tensorboard import program

    tb = program.TensorBoard(plugins,
                             program.get_default_assets_zip_provider())
    argv = _base_tb_args(logdir, reload_interval,
                         path_prefix) + _extra_tb_args(tensorboard_options)
    log.debug("TensorBoard args: %r", argv)
    tb.configure(argv)

    ingester = data_ingester.LocalDataIngester(tb.flags)
    ingester.start()
    return TensorBoardWSGIApp(
        tb.flags,
        tb.plugin_loaders,
        ingester.data_provider,
        tb.assets_zip_provider,
        ingester.deprecated_multiplexer,
    )
示例#4
0
 def run(self):
     tb = program.TensorBoard(
         default.get_plugins() + default.get_dynamic_plugins(), 
         program.get_default_assets_zip_provider())
     tb.configure(logdir = str(self.output / "tb"), port = self.port)
     log = logging.getLogger('tensorflow')
     log.setLevel(logging.CRITICAL)
     tb.main()
示例#5
0
    def run(self):
        # Remove http message
        log = logging.getLogger('werkzeug').setLevel(logging.ERROR)
        # Start tensorboard server
        tb = customTensorBoard(
            default.get_plugins() + default.get_dynamic_plugins(),
            program.get_default_assets_zip_provider())

        tb.configure(
            argv=[None, '--logdir', self.dir_path, '--port',
                  str(self.port)])
        url = tb.launch()
        self.tb = tb
        return url
示例#6
0
def run_main():
  """Initializes flags and calls main()."""
  program.setup_environment()
  tensorboard = program.TensorBoard(get_notf_plugins(),
                                    program.get_default_assets_zip_provider())
  try:
    from absl import app
    # Import this to check that app.run() will accept the flags_parser argument.
    from absl.flags import argparse_flags
    app.run(tensorboard.main, flags_parser=tensorboard.configure)
    raise AssertionError("absl.app.run() shouldn't return")
  except ImportError:
    pass
  tensorboard.configure(sys.argv)
  sys.exit(tensorboard.main())
def main(logdir):
    tb = program.TensorBoard(
        default.get_plugins(),
        program.get_default_assets_zip_provider(),
    )
    tb.configure(bind_all=False, logdir=logdir)
    url = tb.launch()
    sys.stdout.write("TensorBoard started at %s\n" % (url))
    sys.stdout.flush()
    while True:
        try:
            time.sleep(60)
        except KeyboardInterrupt:
            break
    sys.stdout.write("TensorBoard is shutting down")
    sys.stdout.flush()
示例#8
0
 def run(self):
     '''Launch the tensorboard.
     Note that this method would not block the main thread, we
     suggest to use launch() instead of this when you do not need
     to work with subthread.
     '''
     program.setup_environment()
     # Remove http messages
     log = logging.getLogger('werkzeug').setLevel(logging.ERROR)
     # Start tensorboard server
     _tb = program.TensorBoard(
         default.get_plugins(),
         program.get_default_assets_zip_provider())
     _tb.configure(argv=self.__collect_argvs())
     url = _tb.launch()
     print('TensorBoard at {0}, working on path: {1}.'.format(url, self.log_dir))
示例#9
0
def GatherAndServe(logdir: pathlib.Path, hosts: typing.List[str]):
    paths = [str(logdir)] + [f"ssh://{host}{logdir}" for host in FLAGS.hosts]

    with tempfile.TemporaryDirectory(prefix="phd_java_fuzz_") as d:
        working_dir = pathlib.Path(d)
        outdir = GatherAndArrangeLogs(paths, working_dir)

        # Launch tensorboard
        tensorboard = program.TensorBoard(
            default.get_plugins() + default.get_dynamic_plugins(),
            program.get_default_assets_zip_provider(),
        )
        tensorboard.configure(["unused_arg0", "--logdir", str(outdir)])
        tensorboard_url = tensorboard.launch()
        app.Log(1, "Serving tensorboard at %s", tensorboard_url)

        while True:
            GatherAndArrangeLogs(paths, working_dir)
            time.sleep(10)
    def run(self):
        # Remove http messages
        dummy = logging.getLogger('werkzeug').setLevel(logging.ERROR)

        # address the tensorboard "unable to get first event timestamp for run" bug
        events_folders = [ root
                           for root, dirs, files in os.walk(self.dir_path)
                           for name in files
                           if events_filename_pattern.match(name) ]
        #print(str(events_folders))
        for events_folder in events_folders :
            writer = FileWriter( events_folder )
            writer.close()

        # Start tensorboard server
        self.tb = program.TensorBoard(
            default.get_plugins(), program.get_default_assets_zip_provider())
        self.tb.configure(argv=[None, '--logdir', self.dir_path])
        url = self.tb.launch()
        sys.stdout.write('TensorBoard %s at %s [ %s ]\n' %
                         (version.VERSION, url
                          , "http://localhost:" + re.search(host_port_pattern_str, url).group('port')
                         ))