How can i use my own GPU to train
Author: Dan-H-Muniz-SanchezCreated Jun 5, 2024Updated Jul 14, 2025
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.
Source: xuebinqin/U-2-Net