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

How to activate GPU training?

Open usama-muneer opened this issue 1 year ago • 2 comments

I tried to find out the GPU training setting. Can you share the code snippet where can I set the GPU for training and testing?

usama-muneer avatar Jun 03 '23 08:06 usama-muneer