mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Faster
This commit is contained in:
@ -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;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user