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

Validates that all input parameters have the expected dtype.

Parameters:

• named_params (Iterable[Tuple[str, torch.nn.Parameter]]) – Iterable of named parameters.

• dtype (torch.dpython:type) – Expected dtype.

• exclude_param_names (Optional[List[str]]) – Optional list of parameter names to exclude from dtype checking

Raises:

ValueError – If any parameter has a different dtype than dtype.