mmclassification

Author	SHA1	Message	Date
takuoko	43024cda73	[Feature] Support gem pooling (#677 ) * add gem pooling * add example config * fix params * add assert * add param clamp * add test assert * add clamp * fix conflict	2022-02-17 02:17:36 +08:00
Ma Zerun	5de480ea9e	[Feature] Support HRNet and add pre-trained models (#660 ) * Support HRNet * Add HRNet configs * Fix a bug in backward * Add configs and update docs. * Not use bias in conv before batch norm * Defaults to use `norm_eval=False` * Add unit tests and support out_channels in HRFuseScales * Update checkpoint path * Update docstring. * Remove incorrect files * Improve according to comments	2022-01-28 10:54:14 +08:00
Ma Zerun	f9eb9b409b	[Docs] Add Copyright information. (#413 )	2021-08-17 19:52:42 +08:00
Ma Zerun	1a7cebe4b9	[Refactor] Refactor unittest (#321 ) * Refactor unit tests folder structure. * Remove label smooth and Vit test in `test_classifiers.py` * Rename test_utils in dataset to test_dataset_utils * Split test_models/test_utils/test_utils.py to multiple sub files. * Add unit tests of classifiers and heads * Use patch context manager. * Add unit test of `is_tracing`, and add warning in `is_tracing` if torch verison is smaller than 1.6.0	2021-07-08 22:49:05 +08:00