def __init__(self, run=None, comment='default', image_freq=50): View.__init__(self) SummaryWriter.__init__(self, run, comment) self.image_freq = image_freq self.dispatch = { 'tb_step': self.step, 'tb_scalar': self.scalar, 'image': self.image, 'histogram': self.histogram, 'text': self.text } self.global_step = None
def __init__(self, run=None, comment='default', image_freq=100): View.__init__(self) SummaryWriter.__init__(self, run, comment) self.image_freq = image_freq self.global_step = 0 self.dispatch = {'tb_step': self.step, 'tb_scalar':self.scalar, 'image':self.image}