Python Megatron_GPT2示例

示例#1

文件： run_sanity_check.py 项目： wenhuach/DeepSpeed

def test_run():
    runner = unittest.TextTestRunner(failfast=True)

    # Add test suites here.
    pytest_hack(runner.run(Megatron_GPT2.suite()))
    pytest_hack(runner.run(Megatron_GPT2.checkpoint_suite()))
    pytest_hack(runner.run(BingBertSquad.suite()))

示例#2

文件： run_sanity_check.py 项目： szhengac/DeepSpeed

def test_megatron_checkpoint():
    runner = unittest.TextTestRunner(failfast=True)
    pytest_hack(runner.run(Megatron_GPT2.checkpoint_suite()))