mindformers.wrapper

Wrapper

mindformers.wrapper.MFTrainOneStepCell

MindFormers的单步训练包装接口。

mindformers.wrapper.MFPipelineWithLossScaleCell

为MindFormers的单步训练单元扩充流水线并行的损失缩放功能。