mmpretrain/mmcls/datasets
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
..
pipelines [Fix] Fix Albu crash bug. (#918) 2022-07-28 14:10:34 +08:00
samplers [Feature] Add diff seeds to diff ranks. (#744) 2022-03-25 15:31:13 +08:00
__init__.py [Feature] Add `CustomDataset`. (#738) 2022-03-30 15:28:26 +08:00
base_dataset.py [Feature] Add `CustomDataset`. (#738) 2022-03-30 15:28:26 +08:00
builder.py [Fix] Fix lint and mmcv version requirement for IPU. 2022-04-29 22:33:29 +08:00
cifar.py [Feature] Add `CustomDataset`. (#738) 2022-03-30 15:28:26 +08:00
cub.py [Feature] Support CUB dataset. (#703) 2022-03-16 16:22:28 +08:00
custom.py [Feature] Add `CustomDataset`. (#738) 2022-03-30 15:28:26 +08:00
dataset_wrappers.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
imagenet.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
imagenet21k.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
mnist.py
multi_label.py [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
utils.py
voc.py