How to change the parameters in the py file to train gpu?

Question

Opened this issue 5 months ago · 3 comments

I changed two cpu parameters in inference.py to gpu but still didn't call gpu. Which code parameter should I change to complete gpu training?

Answer 1 · 2024-07-15T14:16:48.000Z

I'm sorry I don't understand your question. Could you provide more details?

Answer 2 · 2024-07-16T03:06:45.000Z

I want to use GPU to train this model, but now it's CPU training. What specific location in the code should I change to use my GPU for training

Answer 3 · 2024-07-16T06:08:50.000Z

Please read training part of README