pixelshuffler GPU implementation?

ShenghaiRong / caffe_srgan

A Caffe Implementation of SRGAN

58 stars 22 forks source link

pixelshuffler GPU implementation? #13

Open oldie77 opened 5 years ago

oldie77 commented 5 years ago

Hey there,

your pixelshuffler works very well, thank you for sharing it!! It makes training very slow, though, because it's CPU, only. I don't suppose you have a GPU (reshape_layer.cu) version available?

Thanks!!