Mashiro 399f76ffa8 [Experimental] Add support for FSDP (#1213)

2023-06-29 15:19:33 +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

[Experimental] Add support for FSDP (#1213)

2023-06-29 15:19:33 +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