我尝试修改了训练集路径,然后执行train(),报错如下信息,是否是内存不足引起的?如果是,请问训练CSANMT模型需要多大内存呢?
训练后的模型会保存到哪里呢?
在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
替换为您希望保存模型的路径。
您先用base模型试一下。python3.8/site-packages/modelscope/trainers/nlp/csanmt_translation_trainer.py train()方法中找一下。此回答整理自钉钉群:魔搭ModelScope开发者联盟群 ①