src/network.c
@@ -103,8 +103,8 @@ } 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); //secret_update_connected_layer((connected_layer *)net.layers[i]); update_connected_layer(layer); } } } @@ -372,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; } } }