mmclassification/mmcls/models
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
..
backbones [Fix] Fix Conformer forward with irregular input size. (#686) 2022-02-28 10:17:52 +08:00
classifiers [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
heads [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
losses [Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. (#696) 2022-02-28 10:20:31 +08:00
necks [Feature] Support gem pooling (#677) 2022-02-17 02:17:36 +08:00
utils [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
builder.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00