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/colossalai/engine/gradient_handler
Hakjin Lee 46c009dba4
[format] Run lint on colossalai.engine (#3367)
2 years ago
..
__init__.py [NFC] polish colossalai/engine/gradient_handler/__init__.py code style (#3329) 2 years ago
_base_gradient_handler.py [format] Run lint on colossalai.engine (#3367) 2 years ago
_data_parallel_gradient_handler.py [format] Run lint on colossalai.engine (#3367) 2 years ago
_moe_gradient_handler.py [NFC] polish colossalai/engine/gradient_handler/_moe_gradient_handler.py (#3260) 2 years ago
_pipeline_parallel_gradient_handler.py [format] Run lint on colossalai.engine (#3367) 2 years ago
_sequence_parallel_gradient_handler.py [format] Run lint on colossalai.engine (#3367) 2 years ago
_zero_gradient_handler.py [format] Run lint on colossalai.engine (#3367) 2 years ago
utils.py