mmpretrain/mmcls/models/utils
Ma Zerun 29f066f7fb
[Improve] Speed up data preprocessor. (#1064)
* [Improve] Speed up data preprocessor.

* Add ClsDataSample serialization override functions.

* Add unit tests

* Modify configs to fit new mixup args.

* Fix `num_classes` of the ImageNet-21k config.

* Update docs.
2022-10-17 17:08:18 +08:00
..
batch_augments [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
__init__.py [Feature] Support EdgeNeXt for dev-1.x. (#1037) 2022-09-21 13:27:04 +08:00
attention.py [Feature] Support DeiT3. (#1065) 2022-10-10 14:54:20 +08:00
channel_shuffle.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
data_preprocessor.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
embed.py [Feature] Support Swin Transformer V2 for MMCLS 1.x. (#1029) 2022-09-20 15:45:27 +08:00
helpers.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
inverted_residual.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
layer_scale.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
make_divisible.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
position_encoding.py [Feature] Support EdgeNeXt for dev-1.x. (#1037) 2022-09-21 13:27:04 +08:00
se_layer.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00