IsabelFunke

Thank you & two minor issues

1

Hi, thank you so much for providing this code! Using the automatic computation graph parser, I was able to use the optimal gradient checkpoints during model training without writing much...