From 1ddaaf685a3c5febca32305fc588f982b5e4cdaa Mon Sep 17 00:00:00 2001 From: natanielruiz <nataniel777@hotmail.com> Date: 星期四, 28 九月 2017 06:21:15 +0800 Subject: [PATCH] Final code before submission --- code/train_preangles.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/code/train_preangles.py b/code/train_preangles.py index 4752aef..ffceee2 100644 --- a/code/train_preangles.py +++ b/code/train_preangles.py @@ -125,7 +125,7 @@ if args.dataset == 'Pose_300W_LP': pose_dataset = datasets.Pose_300W_LP(args.data_dir, args.filename_list, transformations) - if args.dataset == 'Pose_300W_LP_random_ds': + elif args.dataset == 'Pose_300W_LP_random_ds': pose_dataset = datasets.Pose_300W_LP_random_ds(args.data_dir, args.filename_list, transformations) elif args.dataset == 'AFLW2000': pose_dataset = datasets.AFLW2000(args.data_dir, args.filename_list, transformations) -- Gitblit v1.8.0