title: Memory saving gradient
body: <p>I work on abuntu 20.04 and I made migration on python program from TF1 to TF2 , this program contains two procedures to run run in CPU successfully (first procedure) but when I want to run with GPU ( second procedure) ,I need to access the file at <a href="https://github.com/cybertronai/gradient-checkpointing" rel="nofollow noreferrer">https://github.com/cybertronai/gradient-checkpointing</a> , program stop after reading the parameters and not training.
  I ask if can any one help me to solve this problem?</p>