def _save_gin(output_dir, sw=None): config_path = os.path.join(output_dir, 'config.gin') config_str = gin.operative_config_str() with gfile.GFile(config_path, 'w') as f: f.write(config_str) if sw: sw.text('gin_config', jaxboard.markdownify_operative_config_str(config_str))
def save_gin(self): config_path = os.path.join(self._output_dir, 'config.gin') config_str = gin.operative_config_str() with tf.io.gfile.GFile(config_path, 'w') as f: f.write(config_str) sw = self._train_sw if sw: sw.text('gin_config', jaxboard.markdownify_operative_config_str(config_str))
def save_gin(self): assert self._output_dir is not None config_path = os.path.join(self._output_dir, 'config.gin') config_str = gin.operative_config_str() with tf.io.gfile.GFile(config_path, 'w') as f: f.write(config_str) if self._sw: self._sw.text('gin_config', jaxboard.markdownify_operative_config_str(config_str))
def _save_gin(self, summary_writer): """"Saves the operative gin config.""" if not self.is_chief or self._output_dir is None: return config_path = os.path.join(self._output_dir, 'config.gin') config_str = gin.operative_config_str() with tf.io.gfile.GFile(config_path, 'w') as f: f.write(config_str) if summary_writer is not None: summary_writer.text( 'gin_config', jaxboard.markdownify_operative_config_str(config_str) )
def save_gin(self): """"Saves the operative gin config, only if it is the chief.""" if not self._is_chief: return assert self._output_dir is not None config_path = os.path.join(self._output_dir, 'config.gin') config_str = gin.operative_config_str() with tf.io.gfile.GFile(config_path, 'w') as f: f.write(config_str) sw = self._train_sw if sw: sw.text('gin_config', jaxboard.markdownify_operative_config_str(config_str))