ModelScope中output路径下没有指定使用具体模型,怎么指定某个训练过程中得到特定模型呢?

ModelScope中output路径下没有指定使用具体哪一个pth模型,怎么指定某个训练过程中得到的特定模型呢?

展开
收起
鸡蛋灌饼儿 2023-01-27 15:35:51 452 分享 版权
1 条回答
写回答
取消 提交回答
  • 目前output中的bin文件,如果是Checkpointhook,那么存储的是最后一个epoch或iter的bin,如果是BestCkptSaverHook,那么存储的是最佳的ckpt,如果想替换其中的bin,可以用torch.load读取pth文件,将其中的‘state_dict’存成bin覆盖到output中——该回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-01-28 07:57:40
    赞同 展开评论

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

收录在圈子:
+ 订阅

热门讨论

热门文章

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