##############################
# filetask
-parser.add_argument("--filetask_file", type=str, default=None)
+parser.add_argument("--filetask_train_file", type=str, default=None)
+
+parser.add_argument("--filetask_test_file", type=str, default=None)
##############################
# rpl options
if args.task == "file":
assert (
- args.filetask_file is not None
- ), "You have to specify the task file with --filetask_file <filename>"
+ args.filetask_train_file is not None and args.filetask_test_file is not None
+ ), "You have to specify the task train and test files"
task = tasks.TaskFromFile(
- args.filetask_file,
+ args.filetask_train_file,
+ args.filetask_test_file,
nb_train_samples=args.nb_train_samples,
nb_test_samples=args.nb_test_samples,
batch_size=args.batch_size,