lts4 / sparsefool Goto Github PK

• windows10
• torch:0.4.1
• torchvision:0.2.1

Hi,I have read your published paper and benefited from it. Thank you very much!
I downloaded the code and ran test_sparsefool.py. I don't know if it's because of the torch version .There is an error：

Traceback (most recent call last):
  File "D:/pythonwork/SparseFool-master/test_sparsefool.py", line 22, in <module>
    net = torch_models.vgg16(pretrained=True)
TypeError: vgg16() got an unexpected keyword argument 'pretrained'

Process finished with exit code 1

I hope I can get your help,Thanks!!!