ed2a46e8c254dde364b760de8357ebf46fe392b6,inferno/trainers/basic.py,Trainer,train_for,#Trainer#Any#Any#,1430

Before Change

            // Update state from model"s state hooks
            self.update_state_from_model_state_hooks()
            // Update parameters
            self.optimizer.step()
            // Call callback
            self.callbacks.call(self.callbacks.END_OF_TRAINING_ITERATION,
                                iteration_num=iteration_num)
            // Prepare for next iteration

After Change

            self.update_state("training_loss", thu.unwrap(loss))
            // Update state from model"s state hooks
            self.update_state_from_model_state_hooks()
            if iteration_num % self.backprop_every == 0:
               // Update parameters
                self.optimizer.step()
                // Zero out the grads
                self.optimizer.zero_grad()
            // Call callback
            self.callbacks.call(self.callbacks.END_OF_TRAINING_ITERATION,
                                iteration_num=iteration_num)
            // Prepare for next iteration
            self.next_iteration()

Instances