Function get_sample_writer

SwissArmyTransformer/sat/training/utils.py:29–33

Returns a tensorboard summary writer

(name, base="..", iteration=0)

SUMMARY_WRITER_DIR_NAME = 'runs'

def get_sample_writer(name, base="..", iteration=0):
    """Returns a tensorboard summary writer
    """
    return SummaryWriter(
        log_dir=os.path.join(base, SUMMARY_WRITER_DIR_NAME, name), purge_step=iteration)

def init_wandb_writer(args):
    """Initialize wandb writer."""