目录

torchtune.utils

杂项

batch_to_device

获取张量的字典(或嵌套字典)并将它们全部设置为同一设备的函数。

get_device

函数,该函数采用可选的设备字符串,在给定计算机和分布式设置的情况下验证它是否正确且可用,并返回一个 .device()

get_logger

获取带有流处理程序的 Logger。

torch_version_ge

检查 torch 版本是否大于或等于给定版本。

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源