mmclassification/mmcls/datasets/pipelines
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
..
__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 [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00