mmpretrain/mmcls/datasets/pipelines
HinGwenWoong 00f0e0d0be
[Fix] Fix Albu crash bug. (#918)
* Fix albu BUG: using albu will cause the label from array(x) to array([x]) and crash the trainning

* Fix common

* Using copy incase potential bug in multi-label tasks

* Improve coding

* Improve code logic

* Add unit test

* Fix typo

* Fix yapf
2022-07-28 14:10:34 +08:00
..
__init__.py [Refactor] Refactor `Resize` transform and add `Pad` transform. (#506) 2021-11-10 16:46:49 +08:00
auto_augment.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
compose.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
formatting.py [Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723) 2022-04-29 22:22:19 +08:00
loading.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
transforms.py [Fix] Fix Albu crash bug. (#918) 2022-07-28 14:10:34 +08:00