#385·U-2-Net

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.