5 Commits

Author	SHA1	Message	Date
Ross Wightman	13565aad50	Add edgenext_base model def & weight link, update to improve ONNX export #1385	2022-08-05 16:58:34 -07:00
Ross Wightman	a45b4bce9a	x and xx small edgenext models do benefit from larger test input size	2022-07-08 10:53:27 -07:00
Ross Wightman	a1cb25066e	Add edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights.	2022-07-07 22:02:57 -07:00
Ross Wightman	dd9b8f57c4	Add feature_info to edgenext for features_only support, hopefully fix some fx / test errors	2022-07-02 15:20:45 -07:00
Ross Wightman	6064d16a2d	Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320 ... * edgenext refactored for torchscript compat, stage base organization * slight refactor of ConvNeXt to match some EdgeNeXt additions * remove use of funky LayerNorm layer in ConvNeXt and just use nn.LayerNorm and LayerNorm2d (permute)	2022-07-01 15:18:42 -07:00