Commit 159fcfeb authored by Mohammad's avatar Mohammad
Browse files

gpt2 zeroshots added to main

parent 8fae1cdd

tasks/main.py

+2 −0
Original line number Diff line number Diff line
@@ -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))