Skip to content

Commit a4516fe

Browse files
committed
Add pretrained weights for FBNet-C
1 parent a72e9b6 commit a4516fe

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ I've leveraged the training scripts in this repository to train a few of the mod
118118
| SE-ResNeXt-26 (32x4d) | 77.104 (22.896) | 93.316 (6.684) | 16.8M | bicubic |
119119
| SE-ResNet-34 | 74.808 (25.192) | 92.124 (7.876) | 22M | bilinear |
120120
| SE-ResNet-18 | 71.742 (28.258) | 90.334 (9.666) | 11.8M | bicubic |
121+
| FBNet-C | 74.830 (25.170 | 92.124 (7.876) | 5.6M | bilinear |
121122
| Single-Path NASNet 1.00 | 74.084 (25.916) | 91.818 (8.182) | 4.42M | bilinear |
122123

123124
### Ported Weights

models/genmobilenet.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def _cfg(url='', **kwargs):
6363
'mobilenetv3_100': _cfg(url=''),
6464
'chamnetv1_100': _cfg(url=''),
6565
'chamnetv2_100': _cfg(url=''),
66-
'fbnetc_100': _cfg(url=''),
66+
'fbnetc_100': _cfg(url='https://www.dropbox.com/s/0ku2tztuibrynld/fbnetc_100-f49a0c5f.pth?dl=1'),
6767
'spnasnet_100': _cfg(url='https://www.dropbox.com/s/iieopt18rytkgaa/spnasnet_100-048bc3f4.pth?dl=1'),
6868
}
6969

@@ -1227,6 +1227,9 @@ def mobilenetv3_100(num_classes, in_chans=3, pretrained=False, **kwargs):
12271227
def fbnetc_100(num_classes, in_chans=3, pretrained=False, **kwargs):
12281228
""" FBNet-C """
12291229
default_cfg = default_cfgs['fbnetc_100']
1230+
if pretrained:
1231+
# pretrained model trained with non-default BN epsilon
1232+
kwargs['bn_eps'] = _BN_EPS_TF_DEFAULT
12301233
model = _gen_fbnetc(1.0, num_classes=num_classes, in_chans=in_chans, **kwargs)
12311234
model.default_cfg = default_cfg
12321235
if pretrained:

0 commit comments

Comments
 (0)