torchtune.utils

杂项

batch_to_device	该函数接收一个张量字典（或嵌套字典），并将其中所有张量设置到同一设备上。
get_device	接受一个可选的设备字符串的函数，根据机器和分布式设置验证其正确性和可用性，并返回device()。
get_logger	获取一个带有流处理器的日志记录器。
torch_version_ge	检查 torch 版本是否大于或等于指定版本。