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", ...
再次祝贺成果并且希望关注一下这个问题
我想我犯了一个很愚蠢的错误并且已经解决了(
很高兴看到这个项目,当我在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", ...
再次祝贺成果并且希望关注一下这个问题