热门
ModelScope中output路径下没有指定使用具体哪一个pth模型,怎么指定某个训练过程中得到的特定模型呢?
目前output中的bin文件,如果是Checkpointhook,那么存储的是最后一个epoch或iter的bin,如果是BestCkptSaverHook,那么存储的是最佳的ckpt,如果想替换其中的bin,可以用torch.load读取pth文件,将其中的‘state_dict’存成bin覆盖到output中——该回答整理自钉群“魔搭ModelScope开发者联盟群 ①”
包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域