mmclassification/mmcls/models/losses
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 [Feature] Add seesaw loss. (#500) 2021-10-27 10:11:59 +08:00
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
focal_loss.py [Enhance] Focal loss for single label tasks (#548) 2021-11-26 12:36:49 +08:00
label_smooth_loss.py [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
seesaw_loss.py [Feature] Add seesaw loss. (#500) 2021-10-27 10:11:59 +08:00
utils.py [Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. (#696) 2022-02-28 10:20:31 +08:00