Closed PingqiLi closed 5 months ago
Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:
To avoid overflow during the training process of squeezenet_1.0, adjust the amp_level in the config to O0.
(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
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)