From 27bb1922495be16eced4541ffef095e3cb140be4 Mon Sep 17 00:00:00 2001 From: Bryan McCann Date: Wed, 9 Jan 2019 00:28:11 +0000 Subject: [PATCH] unique val tasks --- arguments.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/arguments.py b/arguments.py index f5dee6f..8693b46 100644 --- a/arguments.py +++ b/arguments.py @@ -90,7 +90,11 @@ def parse(): if args.model is None: args.model = 'mcqa' if args.val_tasks is None: - args.val_tasks = deepcopy(args.train_tasks) + args.val_tasks = [] + for t in args.train_tasks: + if t not in args.val_tasks: + args.val_tasks.append(t) + if 'imdb' in args.val_tasks: args.val_tasks.remove('imdb') args.world_size = len(args.devices) if args.devices[0] > -1 else -1