__init__.py 339 B

1234567891011
  1. # Copyright (c) Microsoft Corporation.
  2. # SPDX-License-Identifier: Apache-2.0
  3. # DeepSpeed Team
  4. """
  5. Note: please copy webtext data to "Megatron-LM" folder, before running this script.
  6. """
  7. from .run_func_test import GPT2FuncTestCase
  8. from .run_checkpoint_test import GPT2CheckpointTestCase, checkpoint_suite
  9. from .run_func_test import suite