f2754e0656d8cebe1f785f6af100e4ade241a7f8,thinc/model.py,Model,copy,#Model#,445

Before Change

        for name, is_allocated in self._params.items():
            if is_allocated:
                copied.set_param(name, self.get_param(name))
        for name, is_allocated in self._grads.items():
            if is_allocated:
                copied.set_grad(name, self.get_grad(name))
        return copied

    def to_gpu(self, gpu_id: int) -> None:  // pragma: no cover
        Transfer the model to a given GPU device.

After Change

            shims=[shim.copy() for shim in self.shims]
        )
        for name in self.grad_names:
            copied.set_grad(name, self.get_grad(name).copy())
        return copied

    def to_gpu(self, gpu_id: int) -> None:  // pragma: no cover

Instances