Method __update_properties

python/interface/ltp/module.py:61–72

(
        self, device: Optional[torch.device] = None, dtype: Optional[torch.dtype] = None
    )

        return super().half()

    def __update_properties(
        self, device: Optional[torch.device] = None, dtype: Optional[torch.dtype] = None
    ):
        def apply_fn(module):
            if not isinstance(module, BaseModule):
                return
            if device is not None:
                module._device = device
            if dtype is not None:
                module._dtype = dtype

        self.apply(apply_fn)