commonpower.control.logging_utils.loggers.TensorboardLogger
- class TensorboardLogger(log_dir: str, callback: ~stable_baselines3.common.callbacks.BaseCallback = <class 'commonpower.control.logging_utils.callbacks.SafetyCallback'>)[source]
Bases:
BaseLoggerClass for using tensorboard logging in single-agent stable-baselines3 algorithms.
- Parameters:
log_dir (str) – relative path to logging directory
callback (BaseCallback, optional) – object that implements actual logging during training. By defining a customized callback, additional information can be logged (apart from standard metrics like mean_eps_reward)
Methods
finish_loggingget_log_dirHands over the callback so it can be used by the stable-baselines3 internal logging.