This commit is contained in:
Joseph Redmon
2015-01-12 17:27:08 -08:00
parent f26da0ad5c
commit aa5996d58e
13 changed files with 146 additions and 49 deletions

View File

@ -372,6 +372,10 @@ void set_batch_network(network *net, int b)
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;
}
}
}