18963866ca72089649c0004d0e981b8f2c26636f,thinc/extra/wrappers.py,PyTorchWrapper,begin_update,#PyTorchWrapper#Any#Any#,23

Before Change

                self._optimizer.step()
                self._optimizer.zero_grad()
            return dX
        return self.ops.asarray(y_var.data), backward_pytorch

    def _create_optimizer(self, sgd):
        params = self._model.parameters()

After Change

                self._optimizer.step()
                self._optimizer.zero_grad()
            return torch2xp(x_var.grad)
        return torch2xp(y_var), backward_pytorch

    def _create_optimizer(self, sgd):
        params = self._model.parameters()

Instances