pytorch version is very slow compared with matconvnet

cszn / FFDNet

FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoising (TIP, 2018)

https://ieeexplore.ieee.org/abstract/document/8365806/

469 stars 130 forks source link

pytorch version is very slow compared with matconvnet #27

Open kunkun007 opened 3 years ago

kunkun007 commented 3 years ago

I use your matlab version and pytorch for test ffdnet color model, matlab version's time is about 20ms, but pytorch version's time is about 1s. Cuda version is 8.0 and cudnn is 6.0.21. Can you have some ideas about this? What is your testing time and pytorch version?