-
Notifications
You must be signed in to change notification settings - Fork 12
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Failed to prune #1
Comments
thank you. I will check this fault tomorrow. |
@gmayday1997 when the layer conv-->max pool->route, the conv layer need net.layers[i+1] & net.layers[i+2], net.layers[i+2] is route layer, net.layers[i+2].biases is null, prune may make some mistake. |
@erikguo @anhuipl2010 thanks for pointing this. I have fixed this bug, and it works well now.
|
@gmayday1997 u are handsome. do u add sparsity train code for the last update? |
@gmayday1997 your code fopen() function in prune, maybe take some mistake. maybe you can changed it to fopen("xxxx", "wb"); or fopen("xxxx", "w"); |
@anhuipl2010 thanks, I fixed that. 😄 😊 |
@erikguo can you please let me know how you installed this repo. I git cloned this repo,
[ 96%] Building CUDA object CMakeFiles/dark.dir/src/im2col_kernels.cu.o |
Hello,
I build your darknet by build.sh successfully. Then I use 'darknet prune' command to prune my yolo3-spp. But I always failed when calculating pruning to layer 76.
Do I miss some thing ?
Thank you for your help
The text was updated successfully, but these errors were encountered: