mindformers.wrapper

Wrapper

mindformers.wrapper.MFTrainOneStepCell

TrainOneStep For MindFormer.

mindformers.wrapper.MFPipelineWithLossScaleCell

Append a train-one-step cell with loss scale of pipeline parallel for MindFormers.