mindspore-lab / mindcv

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

fix ema and edgenext yaml bugs #703

Closed Songyuanwei closed 1 year ago

Songyuanwei commented 1 year ago

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

Motivation

更新ema,如果梯度溢出时,当不更新loss和梯度的时候,也不进行ema更新。 (Write your motivation for proposed changes here.)

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)