Commit Graph

4 Commits (2060e433c02d80c117635acc3267ab6cb3515824)

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 6688d0669f PyTorch trained MobileNetV3 weights that match/best paper. Update fbnetc weights to improved copy. 2019-06-04 09:51:05 -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