Hi everyone. I’m training my own dataset with Yolov4 using TAO toolkit. It has a issue that, its computations mostly run on CPU. As a result, it is quite slow. Futher, I have trained with another algorithm like yolov3 and detecnet_v2 and their training mostly run on GPU.
I have some following questions:
Is that normal or did I have a mistake somewhere? I have trained yolov4 with TLT toolkit and I met the same problem.
When training stage finishes, can the model still run on GPU with deepstream?
Could you guys help me with these questions? Thanks in advance!