Text2Poster-ICASSP-22 icon indicating copy to clipboard operation
Text2Poster-ICASSP-22 copied to clipboard

Published 20 hours ago • verified publisher icon chuhaojin

你们这个代码不支持gpu？

Open MtYCNN opened this issue 1 year ago • 4 comments

你们的代码为什么运行起来这么慢，生成一张海报要30多秒。看了看基本没用上GPU啊

MtYCNN avatar Mar 22 '23 07:03 MtYCNN

支持gpu的，或许看看是否pytorch版本的问题？在你的环境下运行torch.cuda.is_available()看看是否使用了GPU？

chuhaojin avatar Mar 22 '23 07:03 chuhaojin

测试了GPU可用。pytorch也没问题，也是GPU版本，运行别的程序都没问题。。。但是一跑您这个就很慢 我是用的你们提供的run.sh脚本，我看了一下计算NMS那里特别慢，假如有五个语句，每个语句进行一次NMS的时间是6、7秒，这就要30多秒了。。。。。 但是我测试你们的quick_start.py程序的时候就比较快 请问这是什么原因呢

MtYCNN avatar Mar 22 '23 08:03 MtYCNN

quick_start.py也是基于这个repo代码搭建的，没什么优化。。或许你看看是不是图像的尺寸太大了，如果图像很大的话就会很慢。也可以再检查一下版本问题和环境配置。。

chuhaojin avatar Mar 22 '23 08:03 chuhaojin

quick_start.py也是基于这个repo代码搭建的，没什么优化。。或许你看看是不是图像的尺寸太大了，如果图像很大的话就会很慢。也可以再检查一下版本问题和环境配置。。

你们这个不是默认会resize到300*400分辨率吗，而且我看了一下，感觉和quick_start.py里面下载的图片的尺寸差不多啊。。。。。

MtYCNN avatar Mar 22 '23 09:03 MtYCNN