mmpretrain/tools
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
..
analysis_tools [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
convert_models [Feature] Support VAN. (#739) 2022-04-28 09:35:17 +08:00
deployment [Enhance] Return code 1 if scripts runs failed. (#694) 2022-03-03 10:54:16 +08:00
misc [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
visualizations [Enhance] Enhance vis-pipeline tool. (#604) 2022-03-04 14:40:02 +08:00
dist_test.sh [Feature] Add multi machine dist_train (#734) 2022-03-23 12:03:03 +08:00
dist_train.sh [Feature] Add multi machine dist_train (#734) 2022-03-23 12:03:03 +08:00
kfold-cross-valid.py [Enhance] Use `--a-b` instead of `--a_b` in arguments. (#754) 2022-03-30 19:16:28 +08:00
slurm_test.sh init commit 2020-05-21 21:21:43 +08:00
slurm_train.sh init commit 2020-05-21 21:21:43 +08:00
test.py [Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723) 2022-04-29 22:22:19 +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