Can we use cudnn to speed up this project?

HyeonwooNoh / DPPnet

DPPnet: Image Question Answering using Convolutional Neural Network with Dynamic Parameter Prediction

Other

95 stars 45 forks source link

Can we use cudnn to speed up this project? #5

Open cloud-waiting-for-wind opened 7 years ago

cloud-waiting-for-wind commented 7 years ago

I find that it cost about 48 hours to train 005_train_DPPnet_finetune_cnn.
Can we use cudnn to speed up this project? Have you ever tried to do so ？ If I want to use cudnn, which part do I need to modify?