Fix squeezenet config wtih 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 squeezenet config wtih amp_level O0 #768

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

To avoid overflow during the training process of squeezenet_1.0, adjust the amp_level in the config to O0.

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)