From dd85f02ab0a2b10f8b242a235bc74896f6d47336 Mon Sep 17 00:00:00 2001 From: Rene Ranftl Date: Fri, 5 Jun 2020 14:00:02 +0200 Subject: [PATCH] Missing dependency --- hubconf.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/hubconf.py b/hubconf.py index 42367f18..615b9a9c 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,5 +1,7 @@ dependencies = ['torch'] +import torch + from models.midas_net import MidasNet def MiDaS(pretrained=True, **kwargs):