search

maybeLx / MVSFormerPlusPlus

Codes of MVSFormer++: Revealing the Devil in Transformer’s Details for Multi-View Stereo (ICLR2024)
Apache License 2.0
142 stars 4 forks source link

Error occurs when load the pre-trained mvs model #21

Open Wuuu3511 opened 1 month ago

Wuuu3511 commented 1 month ago

Nice work! When i load the pre-trained mvs model, this error occurs: checkpoint = torch.load("MVSFormerPlusPlus/pretrained_models/model_best.pth", map_location=torch.device('cpu')) File "/public/home/jiangwu/anaconda3/envs/wj-vpamvsnet-38/lib/python3.8/site-packages/torch/serialization.py", line 712, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "/public/home/jiangwu/anaconda3/envs/wj-vpamvsnet-38/lib/python3.8/site-packages/torch/serialization.py", line 1049, in _load result = unpickler.load() File "/public/home/jiangwu/anaconda3/envs/wj-vpamvsnet-38/lib/python3.8/site-packages/torch/serialization.py", line 1042, in find_class return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'base' I would appreciate it if you could help me with this！ Looking forward to your early reply！

ewrfcas commented 1 month ago

This maybe the issue of torch version.