请问下这个训练为什么这么慢，每个batch训练时占用的显存特别小

zhengchuanpan / GMAN

GMAN: A Graph Multi-Attention Network for Traffic Prediction (GMAN, https://fanxlxmu.github.io/publication/aaai2020/) was accepted by AAAI-2020.

Apache License 2.0

418 stars 106 forks source link

Open estimate123 opened 3 years ago

estimate123 commented 3 years ago

你好，请问下我跑这个代码时为什么训练速度特别慢，感觉是一个batch一个batch跑的，显存只占用了306MB，没有并行跑起来，跑一个epoch可能就得跑好几个小时，请问下这是为什么？您训练时遇到这样的问题了吗？十分感谢

WoNiuHu commented 3 years ago

请问这个问题解决了吗？我也碰到这个问题，求指导下

你好，请问下我跑这个代码时为什么训练速度特别慢，感觉是一个batch一个batch跑的，显存只占用了306MB，没有并行跑起来，跑一个epoch可能就得跑好几个小时，请问下这是为什么？您训练时遇到这样的问题了吗？十分感谢

stanli124 commented 3 years ago

pytorch那个版本是cpu跑的，tf你可以注意一下