torchtune.utils

杂项

batch_to_device	一个函数，接收一个张量字典（或嵌套字典），并将它们全部设置到同一个设备上。
get_device	一个函数，接收一个可选的设备字符串，验证其在给定机器和分布式设置下是否正确和可用，并返回一个 device() 对象。
get_logger	获取一个带流处理器的日志记录器。
torch_version_ge	检查 torch 版本是否大于或等于给定版本。
get_world_size_and_rank	一个函数，获取当前进程在默认进程组中的世界大小（即总进程数）和进程编号。