mmpretrain/tests/test_metrics
HumberMe 1214df083d
[Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. (#696)
* some change to mmcls/models/losses/utils.py:convert_to_one_hot()

* fixed problem: line too long

* fixed wrong output shape

* fixed lint PEP8 E128

* fix lint

* fix lint

* add unit tests

Co-authored-by: Ezra-Yu <1105212286@qq.com>
2022-02-28 10:20:31 +08:00
..
test_losses.py [Enhance] Support single-label, softmax, custom eps by asymmetric loss (#609) 2022-01-21 11:35:14 +08:00
test_metrics.py [Enhance] Imporve efficiency of precision, recall, f1_score and support. (#595) 2021-12-13 17:24:59 +08:00
test_utils.py [Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. (#696) 2022-02-28 10:20:31 +08:00