开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

ModelScope如何修改batch_size?

ModelScope如何修改batch_size?

展开
收起
真的很搞笑 2023-12-27 08:04:47 130 0
2 条回答
写回答
取消 提交回答
  • 月移花影,暗香浮动

    要修改ModelScope的batch_size,可以在创建ModelScope实例时设置batch_size参数。例如:

    from mindspore import Model, context
    from mindspore.train.callback import ModelCheckpoint, CheckpointConfig
    from mindspore.nn.metrics import Accuracy
    from mindspore.common.initializer import TruncatedNormal
    from mindspore.ops import operations as P
    import mindspore.dataset as ds
    import mindspore.dataset.text as text
    import numpy as np
    
    context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
    
    # 定义模型
    class Net(Model):
        def __init__(self):
            super(Net, self).__init__()
            self.fc = P.Dense(10, weight_init=TruncatedNormal(std=0.02))
            self.softmax = P.Softmax()
    
        def construct(self, x):
            x = self.fc(x)
            x = self.softmax(x)
            return x
    
    # 创建ModelScope实例并设置batch_size
    batch_size = 32
    model_scope = ModelScope(net, dataset, batch_size=batch_size)
    
    2023-12-29 17:02:37
    赞同 展开评论 打赏
  • image.png
    ,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-12-27 08:50:54
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载