Commit Graph

172 Commits (a54008bd97057149e6c110ddc8887481508ee595)

Author SHA1 Message Date
Ross Wightman 2060e433c0 Add native PyTorch weights for SE-MnasNet aka MnasNet-A1 2019-06-11 08:54:54 -07:00
Ross Wightman 7d17394bdc Add native PyTorch weights for EfficientNet-B0 w/ top-1 > 76.9
* also add pooling details to default cfg for efficiennets so testtimepool wrapper works
2019-06-10 13:11:36 -07:00
Ross Wightman 105d5702d7
Update README.md 2019-06-04 10:23:36 -07:00
Ross Wightman 6688d0669f PyTorch trained MobileNetV3 weights that match/best paper. Update fbnetc weights to improved copy. 2019-06-04 09:51:05 -07:00
Zhun Zhong b32361c9f8
Correct name 2019-06-03 20:33:04 +10:00
Ross Wightman 99122aac1c Replace ResNet-34 default weights with a great result from my experiments. 2019-05-31 10:42:46 -07:00
Ross Wightman 4bb5e9b224 Ported Tensorflow pretrained EfficientNet weights and some model cleanup
* B0-B3 weights ported from TF with close to paper accuracy
* Renamed gen_mobilenet to gen_efficientnet since scaling params go well beyond 'mobile' specific
* Add Tensorflow preprocessing option for closer images to source repo
2019-05-30 17:55:35 -07:00
Ross Wightman 7a6d61566e Add EfficientNet impl, change existing depth_multipler -> channel_multiplier as definitions have been confused 2019-05-29 23:32:52 -07:00
Ross Wightman 99ab1b1276
Update README.md 2019-05-26 18:48:35 -07:00
Ross Wightman c2de3a922b
Update README.md 2019-05-26 18:46:12 -07:00
Ross Wightman 2da0b4dbc1 Add inception_v3 models via torchvision, 4 different pretrained weight choices 2019-05-23 11:07:06 -07:00
Ross Wightman af1a68d2e1
Update README.md 2019-05-20 14:55:56 -07:00
Ross Wightman a4516fe0fb Add pretrained weights for FBNet-C 2019-05-20 11:05:25 -07:00
Ross Wightman a72e9b67ee
Update README.md 2019-05-18 22:27:42 -07:00
Ross Wightman e3377b0409 Add some info to README 2019-05-15 09:54:07 -07:00
Ross Wightman fee607edf6 Mixup implemention in progress
* initial impl w/ label smoothing converging, but needs more testing
2019-05-13 19:05:40 -07:00
Ross Wightman db056d97e2 Add MobileNetV3 and associated changes hard-swish, hard-sigmoid, efficient head, etc 2019-05-10 23:28:13 -07:00
Ross Wightman 0956cd4b66
Update README.md
Add notes for latest mobile model weights...
2019-04-28 18:06:20 -07:00
Ross Wightman 1cf3ea0467
Update README.md 2019-04-21 16:49:00 -07:00
Ross Wightman 073d31a076
Update README.md 2019-04-14 15:19:58 -07:00
Ross Wightman 7ba78aaaeb
Update README.md 2019-04-14 15:14:37 -07:00
Ross Wightman e8e8bce335
Create README.md 2019-04-14 15:10:52 -07:00