Hugh Pu 42c1e916f6 feat(activation_checkpointing): add `non_reentrant_checkpoint` to support inputs require no grad (#4118) | 1 年之前 | |
---|---|---|
.. | ||
__init__.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 年之前 |
checkpointing.py | 42c1e916f6 feat(activation_checkpointing): add `non_reentrant_checkpoint` to support inputs require no grad (#4118) | 1 年之前 |
config.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 年之前 |