mmclassification/tests/test_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
..
test_backbones [Refactor] Use `reset_classifier` to remove head of timm backbones. (#534) 2021-11-16 16:24:55 +08:00
test_utils [Enhance] Enhence SE layer to support custom squeeze channels. (#417) 2021-08-20 13:31:44 +08:00
test_classifiers.py [Fix] Fix a bug when using iter-based runner with 'val' workflow (#542) 2021-11-22 12:06:47 +08:00
test_heads.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
test_neck.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00