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())
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())
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, )
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()
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
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()
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))
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') ))