def run_main(asset_path):
loader = runsenabler_loader.RunsEnablerLoader("some_dir")
plugins = default.get_plugins() + [
paramplot_plugin.ParamPlotPlugin, loader
]
gr_tensorboard = TensorBoard(plugins, lambda: open(asset_path, 'rb'))
gr_tensorboard.configure(sys.argv)
use_filesystem_controller = gr_tensorboard.flags.use_filesystem_controller
original_logdir = pathlib.Path(gr_tensorboard.flags.logdir)
loader.actual_logdir = str(original_logdir)
if use_filesystem_controller:
# Retrieve the actual log directory and replace it in the context with the new logdir
parent_dir = original_logdir.parent
print("logdir provided: " + str(original_logdir))
new_logdir = parent_dir / "temp_dir"
print("creating temporary workspace in " + str(new_logdir))
# Create the temp dir
new_logdir.mkdir(parents=True)
# swap the original logdir for the new one
gr_tensorboard.flags.logdir = str(new_logdir)
try:
sys.exit(gr_tensorboard.main())
finally:
if use_filesystem_controller:
shutil.rmtree(str(new_logdir))
def run_main(asset_path):
plugins = default.get_plugins() + [
paramplot_plugin.ParamPlotPlugin,
runsenabler_loader.RunsEnablerLoader()
]
gr_tensorboard = TensorBoard(plugins, lambda: open(asset_path, 'rb'))
gr_tensorboard.configure(sys.argv)
sys.exit(gr_tensorboard.main())