New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

关于多GPU训练 #125

Open

ZHB2333 opened this issue Jan 11, 2025 · 1 comment

ZHB2333 commented Jan 11, 2025

当我想要更快的训练IRS dehazing时，使用8卡的V100（32G）batchsize从4增加到128此时显存够用，其他yml参数需要修改么，比如说niter或者lr_G等

Owner

Algolzw commented Jan 31, 2025

可以适当增加一下learning rate就行。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment