9 Commits

Author	SHA1	Message	Date
Mashiro	b14c179fad	[Refactor] Refactor ema hook (#804) ... * Refacot ema hook unit test * Refacot ema hook unit test * Enhance test_after_load_checkpoint * refine error messsage * Refine as comment --------- Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Fix unit test	2023-02-22 00:08:38 +08:00
Mashiro	62456217f9	[Feature] Add test time augmentation base model. (#538) ... * First commit * add BaseTestTimeAugModel * Add unit test * move loop logic to test_step * fix ddp test * rename model to module * optim __init__ * Fix as comment * Fix as comment * make val_step should not be called * make tta do not inherit base model * Fix unit test * Enhance docstring * Fix as comment * Fix as comment * minor refine * minor refine * minor refine * fix unit test * minor refine * minor refine * minor refine * minor refine * minor refine * minor refine * fix unit test	2022-10-10 14:29:33 +08:00
Mashiro	6b1b8a3751	[Fix] Fix unit test in windows (#515)	2022-09-13 11:46:21 +08:00
Mashiro	0fb2b8ca8c	[Fix] Fix circle import in EMAHook (#523)	2022-09-09 19:33:27 +08:00
takuoko	a6f5297727	[fix] EMAHook load state dict (#507) ... * fix ema load_state_dict * fix ema load_state_dict * fix for test * fix by review * fix resume and keys	2022-09-09 11:41:12 +08:00
RangiLyu	8d25dbdeda	[Fix] Fix EMAHook trigger train loop and AveragedModel sync buffer. (#467) ... * [Fix] Fix EMAHook trigger train loop init during testing. * fix sync buffer * update ut * fix sync buffer * fix sync buffer	2022-08-26 14:21:56 +08:00
Mashiro	8770c6c7fc	[Refactor] Refactor data flow to make the interface more natural (#468) ... * [Refactor]: modify interface of Visualizer.add_datasample (#365) * [Refactor] Refactor data flow: refine `data_preprocessor`. (#359) * refine data_preprocessor * remove unused BATCH_DATA alias * Fix type hints * rename move_data to cast_data * [Refactor] Refactor data flow: collate data in `collate_fn` of `DataLoader` (#323) * acollate data in dataloader * fix docstring * refine comment * fix as comment * refactor default collate and psedo collate * foramt test file * fix docstring * fix as comment * rename elem to data_item * minor fix * fix as comment * [Refactor] Refactor data flow: `data_batch` argument of `Evaluator.process is a `dict` (#360) * refine evaluator and metric * compatible with new default collate * replace default collate with pseudo * Handle data_batch in metric * fix unit test * fix unit test * fix unit test * minor refine * make data_batch optional make data_batch optional * rename outputs to predictions * fix ut * rename predictions to outputs * fix docstring * fix docstring * fix unit test * make outputs and data_batch to kwargs * fix unit test * keep signature of metric * fix ut * rename pred_sample arguments to data_sample(Visualizer) * fix loop and ut * [refactor]: Refactor model dataflow (#398) * [Refactor] Refactor data flow: refine `data_preprocessor`. (#359) * refine data_preprocessor * remove unused BATCH_DATA alias * Fix type hints * rename move_data to cast_data * refactor model data flow tmp_commt tmp commit * make val_cfg and test_cfg optional * roll back runner * pass test mmdet * fix as comment fix as comment fix ci in DataPreprocessor * fix ut * fix ut * fix rebase main * [Fix]: Fix test val ddp (#462) * [Fix] Fix docstring and type hint of data flow (#463) * Fix docstring of data flow * change signature of hook * fix unit test * resolve conflicts * fix lint	2022-08-24 22:04:55 +08:00
Mashiro	e907931fb8	Fix unit tests (#449)	2022-08-21 14:54:24 +08:00
Zaida Zhou	4637c49640	[Refactor] Rename test_hook to test_hooks (#425)	2022-08-15 11:45:07 +08:00