DeepSpeed
a825f996 - Fix activation checkpoint unit tests for GPU systems (#421)

Commit

4 years ago

Fix activation checkpoint unit tests for GPU systems (#421)

References

#421 - Fix activation checkpoint unit tests for GPU systems

Author

Shaden Smith

Parents

Files2

deepspeed/runtime/activation_checkpointing
- checkpointing.py
tests/unit
- test_activation_checkpointing.py