mmengine/tests/test_model
Mashiro f930b9fe53
Fix docstring format (#1223)
2023-06-30 10:39:19 +08:00
..
test_base_model [Fix] Add PyTorch 2.0 CI and fix unit tests (#1026) 2023-03-29 13:09:23 +08:00
test_wrappers Fix docstring format (#1223) 2023-06-30 10:39:19 +08:00
test_averaged_model.py [Enhancement] Replace warnings.warn with print_log (#961) 2023-03-06 17:25:28 +08:00
test_base_module.py [Fix] Initialize nested modules in ddp which define 'init_weights' method (#1045) 2023-04-05 10:33:24 +08:00
test_model_utils.py [Fix] Add PyTorch 2.0 CI and fix unit tests (#1026) 2023-03-29 13:09:23 +08:00
test_test_aug_time.py Make TTAModel compatible with FSDP (#611) 2022-12-27 21:39:43 +08:00