KeyError: 'weight' in parameters["weight"] when using resnet-style models for inference

[menghao03]

KeyError: 'weight' in parameters["weight"] when using resnet-style models for inference.

Something like this issue in Crypten

[jeffreysijuntan]

Which version of PyTorch are you using? We use PyTorch 1.6 for CrypGPU. This problem may pop up for a newer PyTorch version.