fix continue training from file