TimeMixer icon indicating copy to clipboard operation
TimeMixer copied to clipboard

Low gpu utilization, especially in multi-gpu

Open JasonAlexTan opened this issue 10 months ago • 2 comments

I am experiencing low gpu utilization while running timemixer and timemixer++. I have tested both single-gpu and multi-gpu setups, and the issue persists. The utilization is especially lower in the multi-gpu. Could you help me understand why this may be happening and suggest solutions to improve?

JasonAlexTan avatar Mar 16 '25 17:03 JasonAlexTan

Thank you for your interest in our work. You can try increasing the batch size. Previously, we set the test batch size to 1, referencing the TSLib code, to ensure no data leakage or crossover, which might lead to low GPU utilization. You can try increasing it, which should help you.

kevinliu2000 avatar Apr 10 '25 15:04 kevinliu2000

Thank you for your interest in our work. You can try increasing the batch size. Previously, we set the test batch size to 1, referencing the TSLib code, to ensure no data leakage or crossover, which might lead to low GPU utilization. You can try increasing it, which should help you.

Just wanted to say thanks. Appreciate the help!

JasonAlexTan avatar Apr 17 '25 17:04 JasonAlexTan