torchtune.utils

其他

batch_to_device	一个函数，它接收一个张量字典（或嵌套字典），并将它们全部设置为相同的设备。
get_device	一个函数，它接收一个可选的设备字符串，验证它在给定机器和分布式设置下是否正确且可用，并返回一个 device()。
get_logger	获取一个带有流处理程序的日志记录器。
torch_version_ge	检查 torch 版本是否大于或等于给定版本。