4441480fde64e42a9c4af205bf2ab8003511172e,test.py,,NN,#,9

Before Change

    
    end = time.time()
    for batch_idx, (inputs, targets, indexes) in enumerate(testloader):
        inputs, targets = inputs.cuda(), targets.cuda()
        inputs, targets = Variable(inputs, volatile=True), Variable(targets)
        batchSize = inputs.size(0)
        features = net(inputs)
        net_time.update(time.time() - end)

After Change

import numpy as np

def NN(epoch, net, lemniscate, trainloader, testloader, recompute_memory=0):
    device = "cuda" if torch.cuda.is_available() else "cpu"

    net.eval()
    net_time = AverageMeter()
    cls_time = AverageMeter()

Instances