File size: 313 Bytes
c4b2b37 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import os
from torch.utils.tensorboard import SummaryWriter
class TensorboardSummary(object):
def __init__(self, directory):
self.directory = directory
self.writer = SummaryWriter(log_dir=os.path.join(self.directory))
def add_scalar(self, *args):
self.writer.add_scalar(*args)
|