I had the problem when testing: The checkpoints are loaded successfully but testing images are not getting generated. 