def _initialize_logging(self,
trial_name="",
experiment_dir=None,
upload_dir=None,
trial_config=None):
# TODO(rliaw): In other parts of the code, this is `local_dir`.
if experiment_dir is None:
experiment_dir = os.path.join(DEFAULT_RESULTS_DIR, "default")
self._experiment_dir = os.path.expanduser(experiment_dir)
# TODO(rliaw): Refactor `logdir` to `trial_dir`.
self._logdir = Trial.create_logdir(trial_name, self._experiment_dir)
self._upload_dir = upload_dir
self.trial_config = trial_config or {}
# misc metadata to save as well
self.trial_config["trial_id"] = self.trial_id
self._logger = UnifiedLogger(self.trial_config, self._logdir,
self._upload_dir)
def tf2_logger_creator(config):
logdir = Trial.create_logdir(identifier, local_dir)
logger = tune_logger.UnifiedLogger(config, logdir, loggers=LOGGERS)
# logger = TF2Logger(config, logdir)
return logger