mmpretrain/mmcls/apis
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 [Enchance] Set a random seed when the user does not set a seed. (#554) 2021-12-02 18:09:55 +08:00
inference.py [Fix] Fix a bug when load checkpoints in mulit-GPUs environment. (#527) 2021-11-10 17:54:43 +08:00
test.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
train.py [Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723) 2022-04-29 22:22:19 +08:00