Commit Graph

5 Commits (3e198e63e81d09693db4f3659f0bf7d137369a54)

Author SHA1 Message Date
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
Ezra-Yu 3482521587
[Feature] Add multi machine dist_train (#734)
* add dist_train with multi machines

* add dist_train with multi machines
2022-03-23 12:03:03 +08:00
Ma Zerun 8ef5aeda40
[Enhance] New-style CPU training and inference. (#674)
* [Enhance] New-style CPU training and inference.

* Add version check in CPU training/test
2022-01-30 20:49:54 +08:00
WINDSKY45 b5bd87d7fa
[Docs] Fix some typo in docs. 2022-01-25 12:32:06 +08:00
Ma Zerun 6f25bebe42
[Docs] Update docs. (#600)
* [Docs] The template provides united Docs and OpenMMLab drop-lists in
banner. Remove them from our `conf.py`.

* Move docs of different languages to the same folder
2021-12-15 10:34:34 +08:00