-
Notifications
You must be signed in to change notification settings - Fork 88
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.
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
显存问题 #44
Comments
Title: Video memory problem |
请问原作者是在单卡显存多少的情况下训练的? |
May I ask how the original author trained with the amount of video memory on a single card? |
How can I support Doka training? |
try smaller batch size or image resolution |
Can 3090 be trained |
我使用了48G的显存的显卡,parser.add_argument('-chunk', type=int, default=5 , help='crop volume depth') #原本96
parser.add_argument('-num_sample', type=int, default=1 , help='sample pos and neg')
parser.add_argument('-roi_size', type=int, default=96 , help='resolution of roi') parser.add_argument('-b', type=int, default=1, help='batch size for dataloader')都会爆显存,而太低的参数会导致loss=1 模式崩溃
The text was updated successfully, but these errors were encountered: