8 Commits

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	317ea3e599	Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump.	2021-03-06 12:48:06 -08:00
Ross Wightman	5a8e1e643e	Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.	2021-01-27 22:06:57 -08:00
Ross Wightman	d55bcc0fee	Finishing adding stochastic depth support to BiT ResNetV2 models	2021-01-16 16:32:03 -08:00
Ross Wightman	855d6cc217	More dataset work including factories and a tensorflow datasets (TFDS) wrapper ... * Add parser/dataset factory methods for more flexible dataset & parser creation * Add dataset parser that wraps TFDS image classification datasets * Tweak num_classes handling bug for 21k models * Add initial deit models so they can be benchmarked in next csv results runs	2021-01-15 17:26:20 -08:00
Ross Wightman	20516abc18	Fix some broken tests for ResNetV2 BiT models	2021-01-04 23:21:39 -08:00
Ross Wightman	ce69de70d3	Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models	2020-12-28 16:59:15 -08:00
Ross Wightman	231d04e91a	ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing.	2020-12-28 16:59:15 -08:00