def accumulate(self, learn):
    bs = find_bs(learn.yb)
    self.total += (to_detach(learn.loss.mean()) * bs)
    self.count += bs
def accumulate(self, learn):
    bs = find_bs(learn.yb)
    self.total += (to_detach(learn.loss.mean()) * bs)
    self.count += bs