Closed PingqiLi closed 5 months ago
Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:
Adjust the amp_level from O2 to O0 to avoid overflow happened during the training of squeezenet_1.0.
(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)
(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)
Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:
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)