def __init__(self, *, max_to_keep: int = 4, save_dir: Optional[str] = None) -> None: self.max_to_keep = max_to_keep if save_dir is None: save_dir = os.path.join(get_run_dir(), "checkpoints") self.save_dir = fs.normpath(save_dir)
def __init__(self, scalar: str, *, name: Optional[str] = None, save_dir: Optional[str] = None) -> None: self.scalar = scalar if name is None: name = self.extreme + '-' + scalar.replace('/', '-') self.name = name if save_dir is None: save_dir = os.path.join(get_run_dir(), 'checkpoints') self.save_dir = fs.normpath(save_dir)
def set_run_dir(dirpath: str) -> None: global _run_dir _run_dir = fs.normpath(dirpath) fs.makedir(_run_dir) prefix = '{time}' if dist.size() > 1: prefix += '_{:04d}'.format(dist.rank()) logger.add(os.path.join(_run_dir, 'logging', prefix + '.log'), format=('{time:YYYY-MM-DD HH:mm:ss.SSS} | ' '{name}:{function}:{line} | ' '{level} | {message}'))
def __init__( self, scalar: str, split: str = None, *, name: Optional[str] = None, save_dir: Optional[str] = None, ) -> None: self.scalar = scalar self.split = split if split is not None else "0" if name is None: name = self.extreme + "-" + scalar.replace("/", "-") if split is not None: name += "-" + split self.name = name if save_dir is None: save_dir = os.path.join(get_run_dir(), "checkpoints") self.save_dir = fs.normpath(save_dir)
def __init__(self, load_dir: Optional[str] = None) -> None: if load_dir is None: load_dir = os.path.join(get_run_dir(), 'checkpoints') self.load_dir = fs.normpath(load_dir)
def __init__(self, save_dir: Optional[str] = None) -> None: if save_dir is None: save_dir = os.path.join(get_run_dir(), 'summary') self.save_dir = fs.normpath(save_dir)
def __init__(self, *, save_dir: Optional[str] = None) -> None: if save_dir is None: save_dir = os.path.join(get_run_dir(), 'tensorboard') self.save_dir = fs.normpath(save_dir)