mindspore_lite.ModelGroupFlag
- class mindspore_lite.ModelGroupFlag[源代码]
ModelGroupFlag 类用于构造 ModelGroup 的标签。目前支持以下场景:
ModelGroupFlag.SHARE_WEIGHT ,共享工作空间内存,ModelGroup 的默认构造标签。
ModelGroupFlag.SHARE_WORKSPACE ,共享权重内存,多个模型共享权重(包括常量和变量)内存。
样例:
>>> import mindspore_lite as mslite >>> context = mslite.Context() >>> context.target = ["Ascend"] >>> context.ascend.device_id = 0 >>> context.ascend.rank_id = 0 >>> context.ascend.provider = "ge" >>> model_group = mslite.ModelGroup(mslite.ModelGroupFlag.SHARE_WEIGHT) >>> model0 = mslite.Model() >>> model1 = mslite.Model() >>> model_group.add_model([model0, model1]) >>> model0.build_from_file("seq_1024.mindir", mslite.ModelType.MINDIR, context, "config0.ini") >>> model1.build_from_file("seq_1.mindir", mslite.ModelType.MINDIR, context, "config.ini")