search

hjimce / compress_yolo

对yolo模型进行网络压缩、修剪等实验
Other
50 stars 33 forks source link

关于在GPU下运行 #6

Open chexiaoyu opened 6 years ago

chexiaoyu commented 6 years ago

在运行您的代码时，我发现在GPU下运行不会进入prune_convolutional_layer这个函数。在GPU下运行，更新网络参数是运行update_network_gpu这个函数，而不是update_network这个函数。请问一下可以在GPU上运行吗，还是说需要在network_kernels.cu文件里添加相应的函数？