ModelScope如何修改batch_size?

ModelScope如何修改batch_size?

展开
收起
真的很搞笑 2023-12-27 08:04:47 187 发布于黑龙江 分享
分享
版权
举报
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 举报
    赞同 评论

    评论

    全部评论 (0)

    登录后可评论
  • image.png
    ,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-12-27 08:50:54 举报
    赞同 评论

    评论

    全部评论 (0)

    登录后可评论

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

热门讨论

热门文章

还有其他疑问?
咨询AI助理
AI助理

你好,我是AI助理

可以解答问题、推荐解决方案等