Is single-GPU training supported?
Neptune1008 opened this issue · comments
Neptune1008 commented
Hello author, Thank you for your great work. Can this program complete training with a single GPU?
Shoukang Hu commented
Yes, you can train the model and do the inference with a single GPU.