search

bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。
MIT License
1.56k stars 352 forks source link

gpurtx4060训练后，进行预测，预测时间大概30秒，太慢了 #166

Open mxj009 opened 1 year ago

mxj009 commented 1 year ago

功能都是OK，就是预测太慢，希望可以改进

CCCkr commented 9 months ago

功能都是OK，就是预测太慢，希望可以改进

你好，请问你的cuda和pytorch版本都是多少？我在train的过程中，发现程序根本没有利用上我的4060显卡，但是cuda.available是正常的。