mindspore-lab / mindcv

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

fix: add warmup to squeezenet_1.0 config to make the training be stable. #762

Closed PingqiLi closed 5 months ago

PingqiLi commented 5 months ago

Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:

Motivation

Squeezenet_1.0在以前的配置下不能稳定训练,偶尔会有validation accuracy 1.0的情况。为了使训练稳定,在训练配置的.yaml文件中增加了warmup_epochs=5。 SqueezeNet_1.0 could not be trained stably under the previous configuration, occasionally resulting in a validation accuracy of 1.0%. To stabilize the training, warmup_epochs=5 was added in the training configuration .yaml file.

Test Plan

(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)

Related Issues and PRs

(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)

The-truthh commented 5 months ago

Please change the checkpoint path in squeezenet.py