natanielruiz
2017-09-13 dfa3664a0f56445b023020a0ddb5eedc2780169a
code/batch_testing.py
@@ -62,7 +62,7 @@
    print 'Loading data.'
    transformations = transforms.Compose([transforms.Scale(224),
    transforms.RandomCrop(224), transforms.ToTensor(),
    transforms.CenterCrop(224), transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
    if args.dataset == 'AFLW2000':