how to speed up one epoch

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.

MIT License

2.08k stars 334 forks source link

how to speed up one epoch #17

Closed linzhenyuyuchen closed 3 years ago

linzhenyuyuchen commented 3 years ago

hello,
Thank you for your sharing , this is a great work! When i trained my data with your code, i found a problem. When every epoch starts, it needs to wait for a long time to see the GPU utilization rised up.. What may be the reason for this ?

chengxuanying commented 3 years ago

Hi @linzhenyuyuchen , We have not met this problem. Could you provide a python profile to help find the lag part?

zhengsx commented 3 years ago

Close this issue due to inactivity for a long time. Feel free to reopen it if the problem still exist.