Before modifying dataset loading for data augmentation.

natanielruiz

2017-09-04 7e2652eace4f443b6a01f006b420e91128f0bb7a

 code/train.py

@@ -105,7 +105,7 @@
    transformations = transforms.Compose([transforms.Scale(224),transforms.RandomCrop(224),
                                          transforms.ToTensor()])

    pose_dataset = datasets.Pose_300W_LP_binned(args.data_dir, args.filename_list,
    pose_dataset = datasets.Pose_300W_LP(args.data_dir, args.filename_list,
                                transformations)
    train_loader = torch.utils.data.DataLoader(dataset=pose_dataset,
                                               batch_size=batch_size,