Add NVIDIA apex support and gradient checkpointing to reduce memory footprint#1090
Open
seovchinnikov wants to merge 6 commits intojunyanz:masterfrom seovchinnikov:master
+83-10
Commits
Commits on Jul 9, 2020
Commits on Jul 10, 2020
Add NVIDIA apex support and checkpointing memory optimization (https://pytorch.org/docs/stable/checkpoint.html)
committed
seovchinnikovAdd NVIDIA apex support and checkpointing memory optimization (https://pytorch.org/docs/stable/checkpoint.html)
committedseovchinnikovAdd NVIDIA apex support and checkpointing memory optimization (https://pytorch.org/docs/stable/checkpoint.html)
committedseovchinnikov
Commits on Jul 16, 2020
- committedseovchinnikov