mindformers.wrapper

MindFormers Wrapper API.

mindformers.wrapper.MFTrainOneStepCell	TrainOneStep For MindFormer.
mindformers.wrapper.MFPipelineWithLossScaleCell	Append an train one step cell with loss scale of pipeline parallel for MindFormers.