src/network.c
@@ -218,6 +218,7 @@ state.delta = prev.delta; } layer l = net.layers[i]; if (l.stopbackward) break; l.backward(l, state); } } @@ -356,6 +357,7 @@ }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;