fix: fix the config of squeezenet_1.0 with amp_level O0 - Githubissues

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: fix the config of squeezenet_1.0 with amp_level O0 #769

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:

[x] You have read the Contributing Guidelines on pull requests
[x] Your code builds clean without any errors or warnings
[x] You are using approved terminology
[ ] You have added unit tests

Motivation

Adjust the amp_level from O2 to O0 to avoid overflow happened during the training of squeezenet_1.0.

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)