add fast-depth tutorial

sony · Oct 10, 2024 · 1134601 · 1134601
1 parent 9dec039
commit 1134601
Show file tree

Hide file tree

Showing 2 changed files with 549 additions and 128 deletions.
diff --git a/tutorials/mct_model_garden/models_pytorch/fastdepth/fastdepth.py b/tutorials/mct_model_garden/models_pytorch/fastdepth/fastdepth.py
@@ -114,10 +114,9 @@ def conv_dw(inp, oup, stride, relu6):
         )
 
 class FastDepth(nn.Module, PyTorchModelHubMixin):
-    def __init__(self, output_size):
+    def __init__(self):
 
         super(FastDepth, self).__init__()
-        self.output_size = output_size
         mobilenet = MobileNetBackbone()
 
         for i in range(14):