Function get_sample_writer

bing_bert/turing/utils.py:11–14

Returns a tensorboard summary writer

(name, base="..")

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

class TorchTuple(tuple):
    def to(self, device, non_blocking=False):