mmpretrain/mmcls/datasets
Hu Di b4eefe4794
[Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723)
* implement training and evaluation on IPU

* fp16 SOTA

* Tput reaches 5600

* 123

* add poptorch dataloder

* change ipu_replicas to ipu-replicas

* add noqa to config long line(website)

* remove ipu dataloder test code

* del one blank line in test_builder

* refine the dataloder initialization

* fix a typo

* refine args for dataloder

* remove an annoted line

* process one more conflict

* adjust code structure in mmcv.ipu

* adjust ipu code structure in mmcv

* IPUDataloader to IPUDataLoader

* align with mmcv

* adjust according to mmcv

* mmcv code structre fixed

Co-authored-by: hudi <dihu@graphcore.ai>
2022-04-29 22:22:19 +08:00
..
pipelines [Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723) 2022-04-29 22:22:19 +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 [Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723) 2022-04-29 22:22:19 +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 [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
multi_label.py [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
utils.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
voc.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00