*** ModuleNotFoundError: No module named 'networks.MSBDN-DFF-v1-1'

BookerDeWitt / MSBDN-DFF

The source code of CVPR 2020 paper "Multi-Scale Boosted Dehazing Network with Dense Feature Fusion"

334 stars 69 forks source link

Closed yzfly closed 3 years ago

yzfly commented 3 years ago

How to load model.pkl file?

python test.py --checkpoint model.pkl command failed when load model.pkl

model = torch.load(join(opt.checkpoint, test_list[i]), map_location=lambda storage, loc: storage)

BookerDeWitt commented 3 years ago

If you download the newly released code, plz rename the networks/MSBDN-DFF.py to networks/MSBDN-DFF-v1-1.py.

yzfly commented 3 years ago

That worked, thank you!