natanielruiz
2017-09-04 7e2652eace4f443b6a01f006b420e91128f0bb7a
code/test.py
File was renamed from code/test_resnet_bins.py
@@ -62,7 +62,7 @@
    transformations = transforms.Compose([transforms.Scale(224),
    transforms.RandomCrop(224), transforms.ToTensor()])
    pose_dataset = datasets.AFLW2000_binned(args.data_dir, args.filename_list,
    pose_dataset = datasets.AFLW2000(args.data_dir, args.filename_list,
                                transformations)
    test_loader = torch.utils.data.DataLoader(dataset=pose_dataset,
                                               batch_size=args.batch_size,