search

pytorch / opacus

Training PyTorch models with differential privacy
https://opacus.ai
Apache License 2.0
1.68k stars 337 forks source link

Some issue with loading model with 'weight' as opposed to 'pretrained=True' #635

Open aymuos15 opened 7 months ago

aymuos15 commented 7 months ago
  1. net = resnet18(weights=ResNet18_Weights.IMAGENET1K_V1) #new pytorch style
  2. net = resnet18(pretrained=True) #opacus tutorial/old style

Would there be any reason for the first one to fail? I get an error while running it but everything is fine for the second version.

Is this happening to anyone else?

Thanks!

(love the library!)

HuanyuZhang commented 6 months ago

What is the error message?