mmpretrain/mmcls/models
Ezra-Yu f361bd52e9
[Fix] Fix a bug when using iter-based runner with 'val' workflow (#542)
* add kwargs and default of optimizer in train_step and val_step

* update docstring

* update docstring

* update optional annotation
2021-11-22 12:06:47 +08:00
..
backbones [Refactor] Use `reset_classifier` to remove head of timm backbones. (#534) 2021-11-16 16:24:55 +08:00
classifiers [Fix] Fix a bug when using iter-based runner with 'val' workflow (#542) 2021-11-22 12:06:47 +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