torchtune.utils

杂项

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