fix tf_efficientnet_b3_ap arguments

rwightman / gen-efficientnet-pytorch

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

Apache License 2.0

1.56k stars 214 forks source link

fix tf_efficientnet_b3_ap arguments #31

Closed jzcruiser closed 4 years ago

jzcruiser commented 4 years ago

in both tf_efficientnet_b3_ap and tf_efficientnet_b3_ns methods, the argument "num_classes" and "in_chan" were not passed down to _gen_efficientnet. both arguments are removed to align with other similar methods.

rwightman commented 4 years ago

missed that, thanks