8cab14aa67b78402bb395546ca66e2b927cac222,fid_score.py,,get_activations,#,63

Before Change

    
    model.eval()

    d0 = images.shape[0]
    if batch_size > d0:
        print(("Warning: batch size is bigger than the data size. "
               "Setting batch size to data size"))
        batch_size = d0

After Change

        end = start + batch_size
        
        // real batch of images here
        images = np.array([imread(str(fn)).astype(np.float32) for fn in files[start:end]])
        images = images.transpose((0, 3, 1, 2))
        images /= 255
        
        batch = torch.from_numpy(images).type(torch.FloatTensor)
        batch = Variable(batch, volatile=True)

Instances