High-Resolution-Remote-Sensing-Semantic-Segmentation-PyTorch icon indicating copy to clipboard operation
High-Resolution-Remote-Sensing-Semantic-Segmentation-PyTorch copied to clipboard
verified publisher icon milleniums

训练了一步后会出现gpu现存不足?

Open yuxin7 opened this issue 3 years ago • 3 comments

原始图片大小：24482448 切片大小：512512 batch-size:16 GPU为3090，显存：24G

yuxin7 avatar Oct 10 '22 07:10 yuxin7

image

yuxin7 avatar Oct 10 '22 07:10 yuxin7

batch-size改小就可以

Wzysaber avatar Nov 22 '22 02:11 Wzysaber