Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 313 Bytes
Newer Older
Elton Zheng's avatar
add test model Megatron_GPT2  
Elton Zheng committed
1
2
3
4
5
6
7 
# Copyright (c) 2019, The Microsoft DeepSpeed Team. All rights reserved.
#
# Note: please copy webtext data to "Megatron-LM" folder, before running this script.

from .run_func_test import GPT2FuncTestCase
from .run_checkpoint_test import GPT2CheckpointTestCase, checkpoint_suite
from .run_func_test import suite