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

在ModelScope中,训练后的模型会保存到哪里呢?

我尝试修改了训练集路径,然后执行train(),报错如下信息,是否是内存不足引起的?如果是,请问训练CSANMT模型需要多大内存呢?6d434e501df8a1ecef5bda09a9a10f5b.png
训练后的模型会保存到哪里呢?

展开
收起
崔问问 2023-12-03 22:40:10 92 0
2 条回答
写回答
取消 提交回答
  • 在ModelScope中,训练后的模型会保存到指定的输出路径。您可以在调用train()函数时指定输出路径,例如:

    from modelscope.pipelines import pipeline
    
    # 下载数据集
    dataset = pipeline("your_dataset_name")
    
    # 指定输出路径
    output_path = "your_output_path"
    
    # 训练模型
    model = dataset.train(output_path=output_path)
    

    请将your_dataset_name替换为您要使用的数据集名称,将your_output_path替换为您希望保存模型的路径。

    2023-12-04 14:44:45
    赞同 展开评论 打赏
  • 您先用base模型试一下。python3.8/site-packages/modelscope/trainers/nlp/csanmt_translation_trainer.py train()方法中找一下。此回答整理自钉钉群:魔搭ModelScope开发者联盟群 ①

    2023-12-04 08:03:47
    赞同 展开评论 打赏

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

相关电子书

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