apple / ml-cvnets

CVNets: A library for training computer vision networks
https://apple.github.io/ml-cvnets
Other
1.76k stars 225 forks source link

Bug fix: add `kernel_size` parameter #64

Closed steve2972 closed 3 months ago

steve2972 commented 1 year ago

MobileNet v3 uses 5x5 convolutions in its model architecture. However, the kernel size is not reflected in the Inverse Residual module. Change: added the kernel_size parameter to reflect MobileNetV3 configurations in _make_layer.