2c757bdbd35f07b2bb44cd21ff9124ccf008fe1e,classification/train_classifier.py,,eval_model,#,64

Before Change

        return self.out(output)

def eval_model(niter, model, valid_x, valid_y):
    model.eval()
    N = len(valid_x)
    criterion = nn.CrossEntropyLoss()
    correct = 0.0
    cnt = 0

After Change

def eval_model(niter, model, valid_x, valid_y):
    with torch.no_grad():
        model.eval()
        N = len(valid_x)
        criterion = nn.CrossEntropyLoss()
        correct = 0.0
        cnt = 0

Instances