Closed sageyou closed 4 months ago
Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:
update value of batch_size and learningRate. batch_size =512 would lead to OOM when training on local machine.
(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
update value of batch_size and learningRate. batch_size =512 would lead to OOM when training on local machine.
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)