Load model and resume training

[Ianmcmill]

Hi!

I'm trying to figure out how to resume training. I have checked your other repos but neither does that. I tried

if __name__ == "__main__":
    model = WGAN(lr = 0.0003, silent = False)
    model.load(4)
    while(True):
        model.train()

(4) being the 4th model saved in /models/gen_4.h5 but this just quits to command prompt. I guess it's not as simple as that. Does the train function need some code to get resume functionality? Or does this need checkpoints?