Parameter

[brijraj08]

Hi Ericsun99,
Thanks for sharing the code and model.
I tried using the code you shared and met an error at line 80 of MobileNetV2.py (self.features.append(nn.AvgPool2d(input_size/32))) asking about input parameters to the AvgPool2d function.
Could you please help me in managing the error which says RuntimeError: avg_pool2d(): argument 'kernel_size' (position 2) must be tuple of ints, not float
?

[blueardour]

try add int() , for example: nn.AvgPool2d(int(input_size/32)) ?

[mkwml]

@blueardour nn.AvgPool2d(input_size/32) --> nn.AvgPool2d(input_size//32)