torchtune.utils

杂项

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