tf2_gradient_checkpointing

Gradient checkpointing implementation for TF2 eager mode

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
Memory_prof_results.txt		Memory_prof_results.txt
README.md		README.md
Results_tf_recompute_grad_bug_fix.txt		Results_tf_recompute_grad_bug_fix.txt
Sample_CNN_memory_prof_results_with_ckpt.txt		Sample_CNN_memory_prof_results_with_ckpt.txt
VGG16_memory_prof_results_with_ckpt.txt		VGG16_memory_prof_results_with_ckpt.txt
gradient_checkpoint_profile.py		gradient_checkpoint_profile.py
gradient_checkpointing.py		gradient_checkpointing.py
run_recompute_grad_with_profile.py		run_recompute_grad_with_profile.py
tf_recompute_grad_tutorial.ipynb		tf_recompute_grad_tutorial.ipynb

Provide feedback