pspnet-pytorch icon indicating copy to clipboard operation
pspnet-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

使用自己的数据集训练失败的问题？

Open mmammoth opened this issue 2 years ago • 1 comments

train.py中把batch_size调小之后还是会出现“CUDA out of memory. Tried to allocate 4.16 GiB”的问题，请问是哪里需要调整呢？

mmammoth avatar Oct 04 '22 08:10 mmammoth

调大一般才有这个问题

bubbliiiing avatar Oct 07 '22 12:10 bubbliiiing