Some cleanup and AFLW working

natanielruiz

2017-08-13 e306e86e925e4211c1c2d2f68de45d5e55f3e215

 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.300W_LP(args.data_dir, args.filename_list,
                                transformations)
    train_loader = torch.utils.data.DataLoader(dataset=pose_dataset,
                                               batch_size=batch_size,