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