GMAN icon indicating copy to clipboard operation
GMAN copied to clipboard

Published 20 hours ago • verified publisher icon zhengchuanpan

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

Open estimate123 opened this issue 4 years ago • 2 comments

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

estimate123 avatar Dec 08 '20 12:12 estimate123

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

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

WoNiuHu avatar Jan 12 '21 13:01 WoNiuHu

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

stanli124 avatar Oct 05 '21 13:10 stanli124