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 https://github.com/cybertronai/gradient-checkpointing , program stop after reading the parameters and not training. I ask if can any one help me to solve this problem?
0 Answers
Related