__init__.py 328 B

12345678
  1. # coding=utf-8
  2. # Copyright (c) 2019, The Microsoft DeepSpeed Team. All rights reserved.
  3. #
  4. # Note: please copy webtext data to "Megatron-LM" folder, before running this script.
  5. from .run_func_test import GPT2FuncTestCase
  6. from .run_checkpoint_test import GPT2CheckpointTestCase, checkpoint_suite
  7. from .run_func_test import suite