semantic-segmentation-pytorch icon indicating copy to clipboard operation
semantic-segmentation-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon CSAILVision

Using Single GPU

Open suesuekkim opened this issue 3 years ago • 1 comments

I encountered the issue below. image

Is there any codes for implementing with single GPU?

Thank you.

suesuekkim avatar Feb 10 '22 04:02 suesuekkim

@suesuekkim https://github.com/CSAILVision/semantic-segmentation-pytorch/issues/58#issue-321770424 this helped me. change batch_data = next(iterator) code in train.py

Muratoter avatar May 08 '22 23:05 Muratoter