def load_checkpoint(self): is_load_successful, self.env.current_epoch, self.env.current_step = load_checkpoint( self.config.checkpoint_dir, self.exe, main_program=self.main_program) return is_load_successful
def load_checkpoint(self): is_load_successful, self.env.current_epoch, self.env.current_step, self.best_score = load_checkpoint( self.config.checkpoint_dir, self.exe, main_program=self.main_program) # Revise max_train_steps when incremental training if is_load_successful: self.max_train_steps = self.env.current_step + self.max_train_steps / self.config.num_epoch * ( self.config.num_epoch - self.env.current_epoch + 1) return is_load_successful