[Colab] ModuleNotFoundError: No module named 'torchvision.models.utils'
m1litaru opened this issue · comments
Hello, @zachteed, @jiadeng @heilaw & congratulations on your awesome work!
I encounter the following problem when running the demo to predict a SE3 field from a pair of RGB-D frames.
Do you have any idea how could I solve the problem so I can infer on KITTI dataset?
P.S. @troylhy1991, I read on another issue that you were also inferring on KITTI dataset. How did you manage to solve this issue?
[Solved] Replacing torchvision.models.utils
with torch.hub
in RAFT-3D/raft3d/blocks/resnet.py will solve this issue.