mmclassification

Author	SHA1	Message	Date
Ma Zerun	556fa567a8	[Feature] Support MViT and add checkpoints. (#924 ) * [Feature] Support MViT. * Add MViT configs and docs * Add unit test * Fix unit tests.	2022-08-02 15:20:16 +08:00
huyu	a19c28fe95	[Enhance] Add `get_cat_ids` and `get_gt_labels` to KFoldDataset. (#721 )	2022-03-23 15:57:36 +08:00
Ma Zerun	b39885d953	[Feature] Support K-fold cross-validation (#563 ) * Support to use `indices` to specify which samples to evaluate. * Add KFoldDataset wrapper * Rename 'K' to 'num_splits' accroding to sklearn * Add `kfold-cross-valid.py` * Add unit tests * Add help doc and docstring	2022-01-19 18:32:55 +08:00
Ezra-Yu	4b2128fdbd	[Fix] Fix bug when using `ClassBalancedDataset` (#555 ) * fix dataset wrapper bug and add unit tests * fix dataset wrapper bug and add unit tests * update unit tests * fix lint * align interface to mmdet * update unit tests	2021-11-26 09:40:00 +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