4b0134242f0e79bcdb022623be29e1e7db5445fc,examples/scripts/ray_parallel.py,DataWorker,compute_gradients,#DataWorker#Any#,45

Before Change

            data, target = next(self.data_iterator)
        self.model.zero_grad()
        output = self.model(data)
        loss = F.nll_loss(output, target)
        loss.backward()
        return self.model.get_gradients()

After Change

            data, target = next(self.data_iterator)
        
        guesses, backprop = self.model(data, is_train=True)
        backprop((guesses - target) / target.shape[0])
        return get_model_grads(self.model)

Instances