Solution for resuming training