def _do_begin_fit(self, n_epoch):
    (self.n_epoch, self.loss) = (n_epoch, tensor(0.0))
    self('begin_fit')
def _do_begin_fit(self, n_epoch):
    (self.n_epoch, self.loss) = (n_epoch, tensor(0.0))
    self('begin_fit')