Skip to content

GitLab

Commit 9a5717e3 authored by Shaden Smith's avatar Shaden Smith
Browse files

Update model examples path.

parent 98f5131b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 9 deletions
tests/model/run_sanity_check.py
View file @ 9a5717e3
...@@ -6,17 +6,12 @@ ...@@ -6,17 +6,12 @@
import sys import sys
import unittest import unittest
sys.path.append('../examples/Megatron_GPT2') sys.path.append('../DeepSpeedExamples/Megatron_GPT2')
sys.path.append('../examples/BingBertSquad')
sys.path.append('../examples/QANet-Pytorch')
sys.path.append('../examples/bing_bert')
import os import os
# Import the test cases here. # Import the test cases here.
import Megatron_GPT2 import Megatron_GPT2
import BingBertSquad
import bing_bert
def pytest_hack(runner_result): def pytest_hack(runner_result):
...@@ -38,9 +33,6 @@ def test_run(): ...@@ -38,9 +33,6 @@ def test_run():
# Add test suites here. # Add test suites here.
pytest_hack(runner.run(Megatron_GPT2.suite())) pytest_hack(runner.run(Megatron_GPT2.suite()))
pytest_hack(runner.run(Megatron_GPT2.checkpoint_suite())) pytest_hack(runner.run(Megatron_GPT2.checkpoint_suite()))
#pytest_hack(runner.run(BingBertSquad.suite()))
#pytest_hack(runner.run(bing_bert.checkpoint_suite()))
#pytest_hack(runner.run(bing_bert.pretrain_suite()))
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment