GunhoChoi/Kind-PyTorch-Tutorial

out of memory issue

drinkingcoder opened this issue · 0 comments

I tried to train UNet with Geforce 1080, but there will be a trouble with out of memory cuda error even if I set batch size with 4. Do you have any suggestions?