pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard
verified publisher icon Cadene

inceptionv4.py: adaptiveAvgPoolWidth should be an int and not a tensor

Open Andrechang opened this issue 6 years ago • 0 comments

adaptiveAvgPoolWidth should be an int and not a tensor

put .item() in line 302 of pretrainedmodels/models/inceptionv4.py the following

adaptiveAvgPoolWidth = features.shape[2].item()

Andrechang avatar May 09 '19 15:05 Andrechang