Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

Published 20 hours ago • verified publisher icon aladdinpersson

EfficientNet Error

Open MaxVanDijck opened this issue 4 years ago • 0 comments

Line 82: https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/CNN_architectures/pytorch_efficientnet.py

nn.Conv2d use a kernel size of 3 which is creating models 2-3x larger than the parameter size defined in the EfficientNet: Rethinking Model Structure paper https://arxiv.org/abs/1905.11946v5

Line 451: https://github.com/keras-team/keras/blob/v2.6.0/keras/applications/efficientnet.py#L526-L549 Official Keras repo uses a kernel size of 1 here

MaxVanDijck avatar Sep 23 '21 01:09 MaxVanDijck