开发者社区 > ModelScope模型即服务 > 正文

pretrained_model 在训练的时候怎么设置?是直接加载当前文件夹的pytorch_mod

pretrained_model 在训练的时候怎么设置?是直接加载当前文件夹的pytorch_model.pt了吗??

展开
收起
cuicuicuic 2023-04-23 12:20:03 113 0
1 条回答
写回答
取消 提交回答
  • 默认加载的是cache路径里面的模型,初次下载是拉取model_id对应的模型 如果需要在训练的时候再使用刚刚训练的模型,可以把./work_dir下面的epoch_*.pth替换cache里面的pytorch_model.pt(例如这是我的cache路径, cp ./work_dir/epoch_3.pth /root/.cache/modelscope/hub/damo/cv_convnextTiny_ocr-recognition-general_damo/pytorch_model.pt), 如果只用来推理的话,可以直接参考 from modelscope.pipelines import pipeline from modelscope.utils.constant import Tasks import os ep_num = 3 # 选择模型checkpoint cmd = 'cp {} {}'.format('./work_dir/epoch_%d.pth' % ep_num, './work_dir/output/pytorch_model.pt') # 'work_dir'为configuration中设置的路径,'output'为输出默认路径 os.system(cmd) ocr_recognition = pipeline(Tasks.ocr_recognition, model='./work_dir/output' ) result = ocr_recognition('http://duguang-labelling.oss-cn-shanghai.aliyuncs.com/mass_img_tmp_20220922/ocr_recognition_icdar13.jpg') print(result),此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-04-27 08:15:42
    赞同 展开评论 打赏

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载