validate_expected_param_dtype¶

torchtune.training.validate_expected_param_dtype(named_params: Iterable[Tuple[str, Parameter]], dtype: dtype, exclude_param_names: Optional[List[str]] = None) → None[source]¶

验证所有输入参数是否具有预期的数据类型。

Parameters:

named_params (Iterable[Tuple[str, torch.nn.Parameter]]) – 命名参数的可迭代对象。
dtype (torch.dpython:type) – 预期的数据类型。
exclude_param_names (可选[列表[str]]) – 可选的参数名称列表，用于排除在数据类型检查之外

Raises:

ValueError – 如果任何参数的数据类型与dtype不同。