src/network.c
@@ -103,6 +103,7 @@ } else if(net.types[i] == CONNECTED){ connected_layer layer = *(connected_layer *)net.layers[i]; //secret_update_connected_layer((connected_layer *)net.layers[i]); update_connected_layer(layer); } } @@ -371,6 +372,10 @@ cost_layer *layer = (cost_layer *)net->layers[i]; layer->batch = b; } else if(net->types[i] == CROP){ crop_layer *layer = (crop_layer *)net->layers[i]; layer->batch = b; } } }