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?