def init(self):
    tracker.set_scalar('accuracy.*', True)
    tracker.set_scalar('loss.*', True)
    tracker.set_scalar('ar_loss.*', False)
    hook_model_outputs(self.mode, self.model, 'model')
    self.state_modules = [self.accuracy, self.memory]
def init(self):
    tracker.set_scalar('accuracy.*', True)
    tracker.set_scalar('loss.*', True)
    tracker.set_scalar('ar_loss.*', False)
    hook_model_outputs(self.mode, self.model, 'model')
    self.state_modules = [self.accuracy, self.memory]