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

Before Change

            self.callbacks.call(self.callbacks.BEGIN_OF_TRAINING_ITERATION,
                                iteration_num=iteration_num)
            // Zero out the grads
            self.optimizer.zero_grad()
            // No interrupts while computing - a SIGINT could shoot down the driver if
            // done at the wrong time. Not sure if this has something to do with pinned memory
            with pyu.delayed_keyboard_interrupt():
                // Get batch

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