Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 339 Bytes
Newer Older
aiss's avatar
update v0.9.2  
aiss committed
1
2
3
4
5
6
7 
# Copyright (c) Microsoft Corporation.
# SPDX-License-Identifier: Apache-2.0

# DeepSpeed Team
"""
Note: please copy webtext data to "Megatron-LM" folder, before running this script.
"""
Elton Zheng's avatar
add test model Megatron_GPT2  
Elton Zheng committed
8
9
10
11 

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