DistributedModelParallel.fused_optimizer（TorchRec）
此接口为TorchRec开源接口，非Rec SDK Torch对外接口。此章节介绍使用Rec SDK Torch时调用的TorchRec接口支持的参数范围。
功能描述
返回稀疏表的融合优化器。
函数原型
1 | def fused_optimizer() |
返回值说明
- 成功：返回系数表的优化器。
- 失败：抛出异常。
使用示例
1 2 3 | from torchrec.distributed.model_parallel import DistributedModelParallel model = DistributedModelParallel(XXX) optimizer = model.fused_optimizer |
父主题： 分表接口