Restart training with a new r_max

[bapatist]

Hi! Can one use restart_latest with a different r_max value? I'm guessing the answer is no, just wanted to be sure.

[ilyes319]

Currently that would not work because of checkpointing. A hack might work though.