mindspore-lab / mindcv

A toolbox of vision models and algorithms based on MindSpore
https://mindspore-lab.github.io/mindcv/
Apache License 2.0
235 stars 143 forks source link

加载任意的epoch对应的optimize权重文件进行断点续训 #712

Closed tacyi closed 3 months ago

tacyi commented 1 year ago

If this is your first time, please read our contributor guidelines: https://gitee.com/mindspore/mindspore/blob/master/CONTRIBUTING.md

Is your feature request related to a problem? Please describe. A clear and concise description of what the problem is. Ex. I'm always frustrated when [...] 模型训练断点续训时加载 optimize权重文件时 https://github.com/mindspore-lab/mindcv/blob/main/train.py逻辑中只能加载 最优的optimize权重文件进行断点续训,没办法指定传入某个epoch的optimize权重文件进行断点续训 Describe the solution you'd like A clear and concise description of what you want to happen. 加载任意的epoch对应的optimize权重文件进行断点续训 Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered. 这个需求来源模型看护,非用户需求。提这个需求方便后续追踪,也方便开发者后续或许增加这个需求的逻辑 Additional context Add any other context or screenshots about the feature request here. 比如下图修改可作为参考: