pytorch-segmentation
pytorch-segmentation copied to clipboard
How to activate GPU training?
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?