mmclassification/mmcls/models
Ezra-Yu 87c67a6a79
[Feature] Support positive weights in BCE. (#516)
* add pos-weight in BCE

* mv pos-weight to kwargs
2021-11-10 17:14:12 +08:00
..
backbones [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +08:00
classifiers [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
heads [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
losses [Feature] Support positive weights in BCE. (#516) 2021-11-10 17:14:12 +08:00
necks [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
utils [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +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