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)
