Mashiro 3715fea15b [Refactor] Refactor the unit tests of SyncBuffersHook (#813)		2023-04-28 17:32:30 +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	[Refactor] Refactor the unit tests of SyncBuffersHook (#813)	2023-04-28 17:32:30 +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