[Testing] Remove redundant models in hidet.testing (#44)

* remove redundent models in hidet.testing * format
hidet-org · Dec 25, 2022 · 9f5d20a · 9f5d20a
1 parent f60c3e9
commit 9f5d20a
Show file tree

Hide file tree

Showing 17 changed files with 5 additions and 1,742 deletions.
diff --git a/gallery/getting-started/quick-start.py b/gallery/getting-started/quick-start.py
@@ -36,7 +36,8 @@
 x = torch.randn(1, 3, 224, 224).cuda()
 model = torch.hub.load(
     'pytorch/vision:v0.9.0', 'resnet18', pretrained=True, verbose=False
-).cuda().eval()
+)
+model = model.cuda().eval()
 
 # we should register the hidet backend for pytorch dynamo
 # only need to do this if you import hidet before torch. Otherwise, it is done automatically

diff --git a/gallery/tutorials/optimize-pytorch-model.py b/gallery/tutorials/optimize-pytorch-model.py
@@ -69,7 +69,9 @@
 x = torch.randn(1, 3, 224, 224).cuda()
 model = torch.hub.load(
     'pytorch/vision:v0.9.0', 'resnet18', pretrained=True, verbose=False
-).cuda().eval()
+)
+model = model.cuda().eval()
+
 
 torch.backends.cudnn.allow_tf32 = (
     False  # tf32 would harm the effective precision of torch's results

diff --git a/python/hidet/testing/__init__.py b/python/hidet/testing/__init__.py
@@ -1,4 +1,3 @@
-from . import hidet_models
 from . import utils
 
 from .utils import benchmark_func, check_unary, check_binary, check_torch_unary, check_torch_binary
diff --git a/python/hidet/testing/bench.py b/python/hidet/testing/bench.py
diff --git a/python/hidet/testing/hidet_models/__init__.py b/python/hidet/testing/hidet_models/__init__.py