set_default_dtype¶
- torchtune.training.set_default_dtype(dtype: dtype) Generator[None, None, None][source]¶
用于设置 torch 默认数据类型的上下文管理器。
- 参数:
dtype (torch.dpython:type) – 上下文管理器内部期望的默认数据类型。
- 返回:
用于设置默认数据类型的上下文管理器。
- 返回类型:
ContextManager
示例
>>> with set_default_dtype(torch.bfloat16): >>> x = torch.tensor([1, 2, 3]) >>> x.dtype torch.bfloat16
