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

Before Change

        for key, value in self._params.items():
            params[key] = None if value is None else self.get_param(key)
        grads = {}
        for key, value in self._grads.items():
            grads[key] = None if value is None else self.get_grad(key)
        copied: Model[InT, OutT] = Model(
            self.name,
            self._func,
            init=self._init,

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
        Transfer the model to a given GPU device.

Instances