mmclassification

mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 14:30:11 +08:00

History

[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

…

accuracy.py

[Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723 )

2022-04-29 22:22:19 +08:00

asymmetric_loss.py

[Enhance] Support single-label, softmax, custom eps by asymmetric loss (#609 )

2022-01-21 11:35:14 +08:00

cross_entropy_loss.py

[Feature] Support positive weights in BCE. (#516 )

2021-11-10 17:14:12 +08:00