I think I’ve misunderstood the checkpoint util, thus my example seems to do nothing.
I thought you have to checkpoint one layer somewhere in your model and all layers before this one will be checkpointed.
As it seems you would have to checkpoint all layers together, which makes more sense.
I try to have a look at your code later on, since today is quite busy.