Skip to content

Commit

Permalink
unique val tasks
Browse files Browse the repository at this point in the history
  • Loading branch information
bmccann committed Jan 9, 2019
1 parent 287b911 commit 27bb192
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion arguments.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 27bb192

Please sign in to comment.