`memory_saving_gradients` Not Working in Tensorflow 1.5

Hello Everyone, I’ve opened an issue at cybertronai/gradient-checkpointing, but no one has been able to help yet. It’s Issue #42 (https://github.com/cybertronai/gradient-checkpointing/issues). I would GREATLY appreciate it if someone could provide some wisdom and help me to get the code working. Much appreciated! gitrdonator