You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_ddp
HELSON f69f9bf223
[zero] add chunk init function for users (#1729)
2 years ago
..
test_ddp_ignore_params.py [zero] add chunk init function for users (#1729) 2 years ago
test_ddp_state_dict.py [feature] A new ZeRO implementation (#1644) 2 years ago
test_reducer.py