During the training process, the GPU disconnected. Is there a demo available for rescheduling and resuming the training?