-
Notifications
You must be signed in to change notification settings - Fork 80
Open
Description
When I tried to run this code, the following error occurred:
Fast Gradient Sign Method
Model: inception_v3
Traceback (most recent call last):
File "fgsm.py", line 62, in
out = model(inp)
File "/home/jinping/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/jinping/anaconda3/lib/python3.6/site-packages/torchvision/models/inception.py", line 117, in forward
x = F.avg_pool2d(x, kernel_size=8)
RuntimeError: Given input size: (2048x5x5). Calculated output size: (2048x0x0). Output size is too small at /home/jinping/project/pytorch/aten/src/THCUNN/generic/SpatialAveragePooling.cu:63
Metadata
Metadata
Assignees
Labels
No labels