DCPDN icon indicating copy to clipboard operation
DCPDN copied to clipboard

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

Open 0119tagawa opened this issue 4 years ago • 0 comments

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'))

0119tagawa avatar Jul 05 '21 09:07 0119tagawa