工具

模块

modelopt.torch.utils.cpp_extension	用于加载CPP / CUDA扩展的实用函数。
modelopt.torch.utils.dataset_utils	用于获取样本和不同数据集的前向循环函数的实用函数。
modelopt.torch.utils.distributed	用于使用 torch.distributed 的实用函数。
modelopt.torch.utils.graph	用于计算图的实用函数。
modelopt.torch.utils.import_utils	处理可能不可用的第三方模块的导入错误抑制。
modelopt.torch.utils.list	用于操作列表的工具。
modelopt.torch.utils.logging	用于日志记录的实用函数。
modelopt.torch.utils.network	PyTorch模型的实用函数。
modelopt.torch.utils.perf	用于性能测量的实用函数。
modelopt.torch.utils.random	用于采样的具有确定性、同步种子的随机数生成器。
modelopt.torch.utils.tensor	PyTorch张量的实用函数。

实用函数。