Commit Graph

5 Commits (9d5d4b8df631a826ce5f1b56e6e9c8b3c5ea1973)

Author SHA1 Message Date
Ross Wightman d55bcc0fee Finishing adding stochastic depth support to BiT ResNetV2 models
5 years ago
Ross Wightman 855d6cc217 More dataset work including factories and a tensorflow datasets (TFDS) wrapper
5 years ago
Ross Wightman 20516abc18 Fix some broken tests for ResNetV2 BiT models
5 years ago
Ross Wightman ce69de70d3 Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models
5 years ago
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.
5 years ago