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 |
|
Ross Wightman
|
751b0bba98
|
Add global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool
|
2020-09-02 16:13:47 -07:00 |
|
Ross Wightman
|
88129b2569
|
Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.
|
2020-06-02 21:06:10 -07:00 |
|
Ross Wightman
|
43225d110c
|
Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args.
|
2020-02-18 14:00:26 -08:00 |
|
Ross Wightman
|
6ca0828166
|
Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2
|
2019-11-28 17:43:00 -08:00 |
|
Ross Wightman
|
7b83e67f77
|
Pass drop connect arg through to EfficientNet models
|
2019-11-22 13:27:43 -08:00 |
|
Ross Wightman
|
171c0b88b6
|
Add model registry and model listing fns, refactor model_factory/create_model fn
|
2019-06-23 18:22:16 -07:00 |