Closed lshhan closed 6 months ago
Hello, I would like to ask if the data loading part of the code is performed on the CPU? During my experiment, the CPU utilization has been very high, but the GPU utilization has been very low. What is the reason for this?
Hi, yes, the data loading is performed on CPUs. Then, they are moved to GPUs (https://github.com/noahzn/Lite-Mono/blob/main/trainer.py#L269)
I'm now closing this issue as there is no response.
Hello, I would like to ask if the data loading part of the code is performed on the CPU? During my experiment, the CPU utilization has been very high, but the GPU utilization has been very low. What is the reason for this?