ModelScope mgeo在训练完成后又自动重新开始训练,不停的循环需要改哪里,只训练一次结束?

ModelScope mgeo在训练完成后又自动重新开始训练,不停的循环需要改哪里,只训练一次结束?

展开
收起
青城山下庄文杰 2023-08-25 15:56:53 99 分享 版权
1 条回答
写回答
取消 提交回答
  • 改这个cfg.train.max_epochs ,int(len(train_dataset) / 32) * cfg.train.max_epochs这里的32改成与您设置的batch_size一致。-此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-08-29 15:41:37
    赞同 展开评论

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

热门讨论

热门文章

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