mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fixed classifier for AlexNet and Resnet50

Browse Source
This commit is contained in:
AlexeyAB
2017-09-15 20:31:48 +03:00
parent c48f8c1573
commit 2baa7bde54
4 changed files with 53 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/network.c
View File

@ -356,6 +356,7 @@ int resize_network(network *net, int w, int h)
}else if(l.type == COST){
resize_cost_layer(&l, inputs);
}else{
fprintf(stderr, "Resizing type %d \n", (int)l.type);
error("Cannot resize this type of layer");
}
if(l.workspace_size > workspace_size) workspace_size = l.workspace_size;