Update base_tensorboard.py

Remove incorrect warning message.
This commit is contained in:
Robert Caulk
2023-05-25 16:35:06 +02:00
committed by GitHub
parent d4183b3fcb
commit f647fb342b
@@ -10,8 +10,7 @@ logger = logging.getLogger(__name__)
class BaseTensorboardLogger: class BaseTensorboardLogger:
def __init__(self, logdir: Path, activate: bool = True): def __init__(self, logdir: Path, activate: bool = True):
logger.warning("Tensorboard is not installed, no logs will be written." pass
"Ensure torch is installed, or use the torch/RL docker images")
def log_scalar(self, tag: str, scalar_value: Any, step: int): def log_scalar(self, tag: str, scalar_value: Any, step: int):
return return
@@ -23,8 +22,7 @@ class BaseTensorboardLogger:
class BaseTensorBoardCallback(TrainingCallback): class BaseTensorBoardCallback(TrainingCallback):
def __init__(self, logdir: Path, activate: bool = True): def __init__(self, logdir: Path, activate: bool = True):
logger.warning("Tensorboard is not installed, no logs will be written." pass
"Ensure torch is installed, or use the torch/RL docker images")
def after_iteration( def after_iteration(
self, model, epoch: int, evals_log: TrainingCallback.EvalsLog self, model, epoch: int, evals_log: TrainingCallback.EvalsLog