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

训练问题 #82

Open

zshuo7615 opened this issue May 17, 2024 · 1 comment

Labels

training

zshuo7615 commented May 17, 2024

采用的内存50GB的3090进行单gpu训练，批次设置为1但是依旧出现显存不足的情况，有解决方案吗

Member

xujz18 commented Aug 1, 2024

可以减少Transformer参数的可训练部分。

The trainable parameters of Transformer Blocks in ImageReward can be reduced.

xujz18 added the training label

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