search

hezhangsprinter / DCPDN

Densely Connected Pyramid Dehazing Network (CVPR'2018)
405 stars 112 forks source link

train.py error because of the torch.utils.serialization module lost #60

Open 0119tagawa opened 3 years ago

0119tagawa commented 3 years ago

In myutils/utils.py, from torch.utils.serialization import load_lua should be changed above import torchfile and vgglua = load_lua(os.path.join(model_folder, 'vgg16.t7')) should be changed above vgglua = torchfile.load(os.path.join(model_folder, 'vgg16.t7'))