search

xw-hu / SINet

IEEE Transactions on Intelligent Transportation Systems (TITS), 2019
Other
80 stars 22 forks source link

cudnn #12

Closed zxduan90 closed 5 years ago

zxduan90 commented 6 years ago

想问一下，如果想用cuda8, cudnn6的话，需要对你添加的caroi_pooling等一些新的层进行修改吗？

xw-hu commented 5 years ago

不需要，但可能需要改makefile.config