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