@User :One possible workaround I see would be to save the whole file and load it again as a checkpoint, but I hope there is a more efficient / faster way to do it.