PSENet.pytorch
PSENet.pytorch copied to clipboard
在eval中,在得到分割图后从gpu复制到cpu的时候,我的耗时大概要800ms,为毛会酱紫啊,模型forward都没这么久,调了好久也不知道哪里出问题了,有人遇到类似的问题了吗?
遇到了,在eval 的时候特别慢,基本上一秒一张图片
测试是好慢啊,所以拉长了两个测试的间隔epoch,请问这是什么原因导致的?
pse本身就不是很快 ,gpu拷贝到cpu有可能是因为torch还有些计算没有完成
GPU到CPU的耗时太严重了