Graphormer icon indicating copy to clipboard operation
Graphormer copied to clipboard

Published 20 hours ago • verified publisher icon microsoft

Very slow training

Open decioren opened this issue 2 years ago • 0 comments

I am training graphormer-slim on a custom dataset with 5K graphs and the loss is coming down nicely. However, the GPU utilization is close to zero while the CPU processes (16) are very busy, which means that the CPU is the bottleneck and I could be training much faster...

Is there anyway to make the CPU side faster?

details: graphs in DGL format 50-100 nodes per graph 1-3 edges per node

decioren avatar Jan 03 '23 10:01 decioren