mmengine/tests/test_model
Mashiro 931db99005
[Enhance] Enhance img data preprocessor (#290)
* fix BaseDataPreprocessor

* fix BaseDataPreprocessor

* change device type to torch.device

* change device type to torch.device

* fix cpu method of base model

* Allow ImgDataPreprocessor do not normalize

* remove unnecessary type ignore

* make mean and std optional

* refine docstring
2022-06-09 20:12:15 +08:00
..
test_base_model [Enhance] Enhance img data preprocessor (#290) 2022-06-09 20:12:15 +08:00
test_wrappers [Feature]: add base model, ddp model wrapper and unit test (#268) 2022-06-07 22:13:53 +08:00
test_averaged_model.py [Fix] Fix ema state dict swapping in EMAHook and torch1.5 ut. (#266) 2022-05-30 16:51:06 +08:00
test_base_module.py [Feat]:Add base module (#277) 2022-06-06 10:51:23 +08:00