[GPUNet/TorchHub] Hiding GPUNet dependencies

nv-kkudrynski · nv-kkudrynski · commit 41ca5820a4e5 · 2022-05-19T02:08:00.000+02:00
diff --git a/PyTorch/Classification/GPUNet/configs/gpunet_torchhub.py b/PyTorch/Classification/GPUNet/configs/gpunet_torchhub.py
@@ -14,9 +14,6 @@
 
 import os
 import torch
-from ..models.gpunet_builder import GPUNet_Builder
-from .model_hub import get_configs, MODEL_ZOO_NAME2TYPE_B1
-from timm.models.helpers import load_checkpoint
 
 def nvidia_gpunet(pretrained=True, **kwargs):
     """Constructs a gpunet model (nn.module with additional infer(input) method).
@@ -27,6 +24,11 @@ def nvidia_gpunet(pretrained=True, **kwargs):
         model_math (str, 'fp32'): returns a model in given precision ('fp32' or 'fp16'). Precision fp32 only gpunets
         model_type (str, 'GPUNet-0'): loads selected model type GPUNet-1.... or GPUNet-P1 or DPUNet-D3. Defaults to GPUNet-0
     """
+
+    from ..models.gpunet_builder import GPUNet_Builder
+    from .model_hub import get_configs, MODEL_ZOO_NAME2TYPE_B1
+    from timm.models.helpers import load_checkpoint
+
     modelType = kwargs.get('model_type', 'GPUNet-0')
     print("model_type=", modelType)
 
@@ -63,4 +65,4 @@ def nvidia_gpunet(pretrained=True, **kwargs):
 
     model = model.to("cuda", modelMath)
     
-    return model
+    return model
