EfficientNet.jl icon indicating copy to clipboard operation
EfficientNet.jl copied to clipboard

Published 20 hours ago • verified publisher icon pxl-th

EfficientNetV2

Open gabrielpreviato opened this issue 3 years ago • 0 comments

On this year there were some upgrades in EfficientNet architecture, named as EfficientNetV2.

This PR still need some work, and I'm doing it "on the fly" while I'm travelling, so there are missing things and some tests that are still to be done.

  • [ ] Add depthwise option for MBConv and FusedMBConv
  • [ ] Add stages and channel dict entries for b1˜b* for EfficientNetV2
  • [ ] Add pretrained weights for EfficientNetV2
  • [ ] Add tests
  • [ ] Add docs

gabrielpreviato avatar Dec 31 '21 12:12 gabrielpreviato