FBNetV3 timm trained weights added for b/d/g variants. Update version to 0.5.2 for pypi release.

README.md

@@ -24,9 +24,11 @@ I'm fortunate to be able to dedicate significant time and money of my own suppor
 ## What's New
 
 ### Jan 6, 2022
-* Tried training a few smaller models, two are good so far, more on the way...
+* Version 0.5.2 w/ release to be pushed to pypi. It's been a while since last pypi update and riskier changes will be merged to main branch soon....
+* Tried training a few small / mobile optimized models, a few are good so far, more on the way...
   * `mnasnet_small` - 65.6 top-1
   * `lcnet_100` - 72.1 top-1
+  * `fbnetv3_b/d/g` - 79.1 / 79.7 / 82.0
 * TinyNet models added by [rsomani95](https://github.com/rsomani95)
 * LCNet added via MobileNetV3 architecture
 

timm/models/mobilenetv3.py

@@ -75,9 +75,15 @@ default_cfgs = {
         url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_mobilenetv3_small_minimal_100-922a7843.pth',
         mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD),
 
-    'fbnetv3_b': _cfg(),
+    'fbnetv3_b': _cfg(
-    'fbnetv3_d': _cfg(),
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_b_224-ead5d2a1.pth',
-    'fbnetv3_g': _cfg(),
+        test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)),
+    'fbnetv3_d': _cfg(
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_d_224-c98bce42.pth',
+        test_input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8)),
+    'fbnetv3_g': _cfg(
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/fbnetv3_g_240-0b1df83b.pth',
+        test_input_size=(3, 288, 288), crop_pct=0.95, pool_size=(9, 9)),
 
     "lcnet_035": _cfg(),
     "lcnet_050": _cfg(),

timm/version.py

@@ -1 +1 @@
-__version__ = '0.5.0'
+__version__ = '0.5.2'