mmengine/tests/test_model
Mashiro 317d8f3178
[Fix] Remove auto find_ignored_modules (#1231)
2023-07-03 19:54:54 +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] Remove auto find_ignored_modules (#1231) 2023-07-03 19:54:54 +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