Commit Graph

36 Commits (8b1f2e8e1f4c73f43d9e956c2162884ab319b1a3)

Author SHA1 Message Date
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs.
* improve consistency of model creation helper fns
* add comments to some of the model helpers
* support passing external default_cfgs so they can be sourced from hub
2021-03-16 22:48:26 -07:00
Ross Wightman 68a4144882 Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg. 2021-02-06 16:30:02 -08:00
Ross Wightman 7613094fb5 Add ViT to sotabench 2020-10-22 16:07:22 -07:00
Ross Wightman 4be5b51e0a Missed moving some seresnet -> legacy in sotabench. Check sotabench cache. 2020-09-25 15:51:04 -07:00
Ross Wightman e8ca45854c More models in sotabench, more control over sotabench run, dataset filename extraction consistency 2020-09-24 15:56:57 -07:00
Ross Wightman 9c406532bd Add EfficientNet-EdgeTPU-M (efficientnet_em) model trained natively in PyTorch. More sotabench fiddling. 2020-09-23 17:12:07 -07:00
Ross Wightman 3681c5c4dd Another sotabench.py debug iter 2020-09-18 23:46:46 -07:00
Ross Wightman 08029852d9 Sotabench debugging 2020-09-18 22:14:51 -07:00
Ross Wightman f225ae8e59 Update README with model results and attribution. Make scheduler factory bit more robust to arg differences, add noise to plateau lr and fix min/max. 2020-06-12 13:33:12 -07:00
Ross Wightman 6c7932fe75 Update sotabench.py, tweak VovNet cfg 2020-06-11 14:34:25 -07:00
Ross Wightman d79ac48626 Update sotabench.py 2020-05-22 14:42:43 -07:00
Ross Wightman ef30e7f2b6 Update sotabench with EffNet Lite models 2020-03-18 13:15:54 -07:00
Ross Wightman c16f25ced2 Add MobileNetV3 Large weights, results, update README and sotabench for merge 2020-02-29 20:37:20 -08:00
Ross Wightman d77f45a6f6 Forgot to add skresnet34 to sotabench 2020-02-18 15:11:41 -08:00
Ross Wightman 569419b38d Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch 2020-02-15 21:18:25 -08:00
Ross Wightman b9496995d8 Add L2-475 PyTorch preprocessing result, update sotabench for new weights 2020-02-12 13:20:37 -08:00
Ross Wightman cade829105 Add EfficientNet-ES to sotabench 2020-02-09 11:33:32 -08:00
Ross Wightman fd98fb33c5 Update sotabench with tf_efficientnet_b8 model 2020-02-01 18:12:25 -08:00
Ross Wightman e42503e1d0 Update sotabench 2020-01-12 18:00:12 -08:00
Ross Wightman 19d93fe454 Add selecsls60 weights 2019-12-31 16:49:04 -08:00
Ross Wightman 84ca3d1f4d Add SelecSLS to sotabench list 2019-12-30 16:04:39 -08:00
Ross Wightman a4497afa21
Update sotabench.py 2019-12-28 23:06:27 -08:00
Ross Wightman 8894a6bc08 Update sotabench.py with latest seresnext26 and efficientnet_b3 models 2019-12-28 17:25:53 -08:00
Ross Wightman 3bef524f9c Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
Ross Wightman 7b3c235ccc Update sotabench with new models 2019-11-28 17:56:59 -08:00
Ross Wightman 0d58c50fb1 Add TF RandAug weights for B5/B7 EfficientNet models. 2019-10-30 16:49:17 -07:00
Ross Wightman 62105ed4a0 Better differentiate sotabench WSL, SSL, and SWSL models via model_desc 2019-10-20 13:17:25 -07:00
Ross Wightman b93fcf0708 Add Facebook Research Semi-Supervised and Semi-Weakly Supervised ResNet model weights. 2019-10-19 17:05:37 -07:00
rwightman d3ba34ee7e Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup. 2019-09-05 15:47:56 -07:00
rwightman b5a8bb52fd Silly typos. 2019-09-04 21:36:58 -07:00
rwightman 7dc5d7a997 sotabench fail 2019-09-04 19:15:36 -07:00
rwightman adbf770f16 Add Res2Net and DLA models w/ pretrained weights. Update sotabench. 2019-09-04 17:06:42 -07:00
Ross Wightman 5c6da1c55a Experiment with sotabench model_description field. 2019-08-28 21:34:39 -07:00
Ross Wightman 81875d52a6 Update sotabench model list, add Mean-Max pooling DPN variants, disable download progress 2019-08-27 00:07:32 -07:00
Ross Wightman 9816ca3ab4 Add EdgeTPU models to sotabench, remove some unicode chars from model names 2019-08-24 14:42:51 -07:00
Ross Wightman daeaa113e2 Add initial sotabench attempt. Split create_transform out of create_loader. Update requirements.txt 2019-08-12 23:06:19 -07:00