Closed 98mxr closed 2 years ago
很高兴看到这个项目，当我在test时，从Google Drive下载的pretrained model参数似乎对应不上。
netNetwork.load_state_dict(torch.load('./model_best.pkl'))
下载下来的文件为model.pkl，且提示
RuntimeError: Error(s) in loading state_dict for M2M_PWC: Missing key(s) in state_dict: "paramAlpha", "netFlow.netExtractor.netOne.netMain.0.weight", "netFlow.netExtractor.netOne.netMain.0.bias", "netFlow.netExtractor.netOne.netMain.1.weight", "netFlow.netExtractor.netOne.netMain.2.weight", "netFlow.netExtractor.netOne.netMain.2.bias", ...
再次祝贺成果并且希望关注一下这个问题
我想我犯了一个很愚蠢的错误并且已经解决了（
98mxr
commented
2 years ago 98mxr
commented
2 years ago
很高兴看到这个项目，当我在test时，从Google Drive下载的pretrained model参数似乎对应不上。
netNetwork.load_state_dict(torch.load('./model_best.pkl'))下载下来的文件为model.pkl，且提示
RuntimeError: Error(s) in loading state_dict for M2M_PWC: Missing key(s) in state_dict: "paramAlpha", "netFlow.netExtractor.netOne.netMain.0.weight", "netFlow.netExtractor.netOne.netMain.0.bias", "netFlow.netExtractor.netOne.netMain.1.weight", "netFlow.netExtractor.netOne.netMain.2.weight", "netFlow.netExtractor.netOne.netMain.2.bias", ...再次祝贺成果并且希望关注一下这个问题