Method to

python/interface/ltp/module.py:32–35

(self, *args, **kwargs)

        raise RuntimeError("Cannot set the device explicitly. Please use module.to(new_device).")

    def to(self, *args, **kwargs) -> Module:
        out = torch._C._nn._parse_to(*args, **kwargs)
        self.__update_properties(device=out[0], dtype=out[1])
        return super().to(*args, **kwargs)

    def cuda(self, device: Optional[int] = None) -> Module:
        self.__update_properties(device=torch.device("cuda", index=device))