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

modelscope里面 训练, 怎么修改配置文件啊?

"model_id = 'damo/mgeo_backbone_chinese_base'
finetune(
model_id=model_id,
train_dataset=train_dataset,
eval_dataset=dev_dataset,
cfg_modify_fn=cfg_modify_fn,
name='nlp-base-trainer')

请教下, modelscope里面 训练, 怎么修改配置文件啊?比如我想修改epoch
"

展开
收起
真的很搞笑 2023-10-19 12:30:06 117 0
3 条回答
写回答
取消 提交回答
  • 1,你想要修改配置文件,可以使用cfg_modify_fn参数。在上述代码中,cfg_modify_fn是一个函数,用于修改配置文件。你可以在该函数中添加你需要修改的配置项,例如修改epoch。

    例如,如果你想修改epoch,可以将以下代码添加到cfg_modify_fn函数中:

    cfg.train.max_epoch = 10

    在上述代码中,cfg.train.max_epoch = 10将训练的最大epoch数修改为10。

    2,你想要重写finetune函数,可以创建一个新的函数,然后在该函数中添加你需要的参数。然后,你可以调用新的函数,将原有的参数传递给新的函数。这样,你就可以在新的函数中修改参数,而不会影响原有的函数。

    2023-10-19 14:25:16
    赞同 展开评论 打赏
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    根据提供的信息,您需要在cfg_modify_fn函数中修改配置文件。具体来说,您可以在cfg_modify_fn函数中使用cfg的merge_from_dict方法,将您想要修改的配置项添加到cfg对象中。例如,如果您想修改epoch参数,可以在cfg_modify_fn函数中添加以下代码:

    1. 创建一个字典,包含epoch参数的值:
    epoch = 10
    cfg_dict = {'epoch': epoch}
    
    1. 使用cfg的merge_from_dict方法,将cfg_dict中的配置项添加到cfg对象中:
    cfg.merge_from_dict(cfg_dict)
    

    关于您的第二个问题,如果要重写finetune函数,建议在finetune函数的基础上进行扩展和修改,以便添加您想要的功能。在修改finetune函数时,建议保留原有的参数,并在函数中进行检查和处理。

    2023-10-19 13:56:55
    赞同 展开评论 打赏

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

相关电子书

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