def __call__(self, batch):
    losses = {}
    (x, y) = batch[0]
    losses['acc_s'] = losses['ce'] = self._predict(x, y)
    with torch.no_grad():
        (x, y) = batch[1]
        losses['acc_t'] = self._predict(x, y)
    return losses
def __call__(self, batch):
    losses = {}
    (x, y) = batch[0]
    losses['acc_s'] = losses['ce'] = self._predict(x, y)
    with torch.no_grad():
        (x, y) = batch[1]
        losses['acc_t'] = self._predict(x, y)
    return losses