papermsucode / 2020bboxrestoration Goto Github PK

Hi, thank you for sharing these codes.

However, I tried to run face_recovery.ipnyb and the following error:

UnpicklingError Traceback (most recent call last)
in
1 net = LResNet100().eval().to(torch.device("cpu"))
----> 2 net.load_state_dict(torch.load("torchtest.pt"))

~/anaconda3/lib/python3.8/site-packages/torch/serialization.py in load(f, map_location, pickle_module, **pickle_load_args)
591 return torch.jit.load(opened_file)
592 return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
--> 593 return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
594
595

~/anaconda3/lib/python3.8/site-packages/torch/serialization.py in _legacy_load(f, map_location, pickle_module, **pickle_load_args)
760 "functionality.")
761
--> 762 magic_number = pickle_module.load(f, **pickle_load_args)
763 if magic_number != MAGIC_NUMBER:
764 raise RuntimeError("Invalid magic number; corrupt file?")

UnpicklingError: invalid load key, 'v'.

I found out that torchtest.pt, which should be the pretrained weight, is only 134 bytes.

I think you uploaded the wrong file.

It would be very nice if you could re-upload the weight file.

Thank you and have a great day :)

Hi, the torchtest.pt file in this repository doesn't seem containing the weights for the model. Could you please release the pre-trained model on google drive or any other way?