How can i use my own GPU to train

xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

Apache License 2.0

8.52k stars 1.47k forks source link

Open Dan-H-Muniz-Sanchez opened 3 months ago

Dan-H-Muniz-Sanchez commented 3 months ago

im not sure if by default u2net_train.py use my GPU or not , how can i be sure? and if the train not use the GPU cow can i change that? Thanks btw.