gpt2 zeroshots added to main

159fcfeb · Mohammad · 8fae1cdd · 159fcfeb
Commit 159fcfeb authored Apr 01, 2020 by Mohammad
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tasks/main.py tasks/main.py +2 -0

No files found.
--- a/tasks/main.py
+++ b/tasks/main.py
@@ -60,6 +60,8 @@ if __name__ == '__main__':
        from race.finetune import main
    elif args.task in ['MNLI', 'QQP']:
        from glue.finetune import main
+    elif args.task in ['LAMBADA', 'WIKITEXT103']:
+        from zeroshot_gpt2.evaluate import main
    else:
        raise NotImplementedError('Task {} is not implemented.'.format(
            args.task))