def _get_file_writer(self): """Returns the default FileWriter instance. Recreates it if closed.""" if self.rank > 0: return TTDummyFileWriter() if self.all_writers is None or self.file_writer is None: if self.purge_step is not None: most_recent_step = self.purge_step self.file_writer = FileWriter(self.log_dir, self.max_queue, self.flush_secs, self.filename_suffix) self.file_writer.debug = self.debug self.file_writer.rank = self.rank self.file_writer.add_event( Event(step=most_recent_step, file_version='brain.Event:2')) self.file_writer.add_event( Event(step=most_recent_step, session_log=SessionLog(status=SessionLog.START))) else: self.file_writer = FileWriter(self.log_dir, self.max_queue, self.flush_secs, self.filename_suffix) self.all_writers = { self.file_writer.get_logdir(): self.file_writer } return self.file_writer
def _get_file_writer(self): """Returns the default FileWriter instance. Recreates it if closed.""" if self.all_writers is None or self.file_writer is None: if 'purge_step' in self.kwargs.keys(): most_recent_step = self.kwargs.pop('purge_step') self.file_writer = FileWriter(logdir=self.log_dir, **self.kwargs) self.file_writer.add_event( Event(step=most_recent_step, file_version='brain.Event:2')) self.file_writer.add_event( Event(step=most_recent_step, session_log=SessionLog(status=SessionLog.START))) else: self.file_writer = FileWriter(logdir=self.log_dir, **self.kwargs) self.all_writers = {self.file_writer.get_logdir(): self.file_writer} return self.file_writer