Source code for draugr.torch_utilities.writers.tensorboard.launcher
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
__author__ = "Christian Heider Nielsen"
__doc__ = ""
__all__ = ["launch_tensorboard"]
from pathlib import Path
from tensorboard import program # IMPORT OUT HERE; POSSIBLE RAISE CONDITIONS
from draugr.python_utilities.sockets import is_port_in_use
[docs]def launch_tensorboard(log_dir: Path, port: int = 6006) -> str:
"""
:param log_dir:
:type log_dir:
:param port:
:type port:
:return:
:rtype:"""
if is_port_in_use(port):
raise RuntimeError(f"Port {port} is in use")
tb = program.TensorBoard()
# tb.configure(argv=['', '--logdir', log_dir, '--port', port])
tb.configure(logdir=str(log_dir), port=port)
tb._fix_mime_types()
return tb.launch()