蒸馏
模块
蒸馏模式的配置。 |
|
用于将模型转换为modelopt.torch.distill.DistillationModel的API,可直接用于训练。 |
|
元模型包装器,用于支持知识蒸馏学习。 |
|
蒸馏任务的基本损失平衡器。 |
|
不同类型的蒸馏损失。 |
|
模块实现并描述了在NAS转换过程中可以使用的模式。 |
|
处理第三方模块的蒸馏插件。 |
|
一个动态模块注册表,用于存储与蒸馏相关的可用DynamicModules的信息。 |
用于torch的蒸馏API子包。
模块
蒸馏模式的配置。 |
|
用于将模型转换为modelopt.torch.distill.DistillationModel的API,可直接用于训练。 |
|
元模型包装器,用于支持知识蒸馏学习。 |
|
蒸馏任务的基本损失平衡器。 |
|
不同类型的蒸馏损失。 |
|
模块实现并描述了在NAS转换过程中可以使用的模式。 |
|
处理第三方模块的蒸馏插件。 |
|
一个动态模块注册表,用于存储与蒸馏相关的可用DynamicModules的信息。 |
用于torch的蒸馏API子包。