Visualizing-CNNs-for-monocular-depth-estimation icon indicating copy to clipboard operation
Visualizing-CNNs-for-monocular-depth-estimation copied to clipboard

Published 20 hours ago • verified publisher icon JunjH

Miss match in pretrained models architecture and given base model architecture

Open abdulkampli opened this issue 4 years ago • 0 comments

Hi JunjH,

while testing the Kitti dataset I am getting this error : Traceback (most recent call last): File "/content/visualize_kitti/test.py", line 123, in main() File "/content/visualize_kitti/test.py", line 76, in main model.load_state_dict(state_dict) File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 1224, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.E.base.0.conv1.weight", "module.E.base.0.bn1.weight", "module.E.base.0.bn1.bias",

could you please share the correct base model architecture for eg. resnet.py

Mnay Thanks, Abdul Capture

abdulkampli avatar Apr 11 '21 15:04 abdulkampli