mmpretrain/mmcls/models/heads
WRH 44bbc71e14
Fix bug and optimize MNIST config (#98)
* add simple_test to ClsHead

* optimize lenet training config

* recover path setting
2020-11-26 15:27:04 +08:00
..
__init__.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
base_head.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
cls_head.py Fix bug and optimize MNIST config (#98) 2020-11-26 15:27:04 +08:00
linear_head.py Add model inference (#16) 2020-09-30 19:00:20 +08:00