def setup_logging_dir(self): """ Sets up the directory where logs will be stored and config files written :return: full path of logging dir :rtype: str """ if 'logging_dir_name' in self._config['training']: dir_name = self._config['training']['logging_dir_name'] else: dir_name = utils.get_current_time_unique_name() +"_" + str(self._config['dense_correspondence_network']['descriptor_dimension']) + "d" self._logging_dir_name = dir_name self._logging_dir = os.path.join(utils.convert_to_absolute_path(self._config['training']['logging_dir']), dir_name) if os.path.isdir(self._logging_dir): shutil.rmtree(self._logging_dir) if not os.path.isdir(self._logging_dir): os.makedirs(self._logging_dir) # make the tensorboard log directory self._tensorboard_log_dir = os.path.join(self._logging_dir, "tensorboard") if not os.path.isdir(self._tensorboard_log_dir): os.makedirs(self._tensorboard_log_dir) return self._logging_dir
def setup_logging_dir(self): """ Sets up the directory where logs will be stored and config files written :return: full path of logging dir :rtype: str """ if 'logging_dir_name' in self._config['training']: dir_name = self._config['training']['logging_dir_name'] else: dir_name = utils.get_current_time_unique_name() +"_" + str(self._config['dense_correspondence_network']['descriptor_dimension']) + "d" self._logging_dir_name = dir_name self._logging_dir = os.path.join(utils.convert_to_absolute_path(self._config['training']['logging_dir']), dir_name) if os.path.isdir(self._logging_dir): shutil.rmtree(self._logging_dir) if not os.path.isdir(self._logging_dir): os.makedirs(self._logging_dir) # make the tensorboard log directory self._tensorboard_log_dir = os.path.join(self._logging_dir, "tensorboard") if not os.path.isdir(self._tensorboard_log_dir): os.makedirs(self._tensorboard_log_dir) return self._logging_dir