53531a5b2d07790c18ed18a54604d977cbdb1189,inferno/net.py,NeuralNet,train_step,#NeuralNet#Any#Any#Any#,468

Before Change

        applied).

        
        xi, yi = to_var(xi), to_var(yi)
        self.module_.train()

        optimizer.zero_grad()

After Change

        
        self.module_.train()
        xi = to_var(xi, use_cuda=self.use_cuda)
        yi = to_var(yi, use_cuda=self.use_cuda)

        optimizer.zero_grad()
        y_pred = self.infer(xi)
        loss = self.get_loss(y_pred, yi, train=True)

Instances