BookerDeWitt/MSBDN-DFF

model.pkl unpacking problem

Opened this issue · 2 comments

Hi,
I want to take a look at what's in model.pkl. However, when I open it using the following code, the output is a series of number.

import pickle
info = pickle.load(open('./model/MSBDNet/checkpoint/msbdn_model.pkl', 'rb'), encoding='utf-8')
print(info)

Output:
119547037146038801333356

Is that the decoding problem? Or do I need some specific package to unpack it, if so,could you please tell which one should I use?
Thank you very much.

Hi @zsjhhh
If you have a look at the README, it states that model.pkl is the checkpoints file.
This means that it contains the weights of the model architecture.
You can load it similar to the line given in test.py.

model = torch.load(test_pkl, map_location=lambda storage, loc: storage)

However, looking at the contents of it won't give you any information.
Thanks

hello, sorry to bother you. when I load the pretrained model in another project like this:
model = torch.load(test_pkl, map_location=lambda storage, loc: storage), the error infomation is "ModuleNotFoundError: No module named 'networks'" , I am puzzled about why the structure should be the same as the original project, with MSBDN-DFF-v1-1.py in the networks directory. Can you help me about this problem? Thank you for your help in advance.