mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

getting rid of sub_arrays, nvidia driver memory leak

Browse Source
This commit is contained in:
Joseph Redmon
2014-10-27 19:45:06 -07:00
parent edbccdfcaf
commit af4e4f92dc
13 changed files with 361 additions and 292 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/network.c
View File

@ -418,7 +418,25 @@ float train_network_sgd_gpu(network net, data d, int n)
int i;
float sum = 0;
for(i = 0; i < n; ++i){
get_batch(d, batch, X, y);
get_random_batch(d, batch, X, y);
float err = train_network_datum_gpu(net, X, y);
sum += err;
}
free(X);
free(y);
return (float)sum/(n*batch);
}
float train_network_data_gpu(network net, data d, int n)
{
int batch = net.batch;
float *X = calloc(batch*d.X.cols, sizeof(float));
float *y = calloc(batch*d.y.cols, sizeof(float));
int i;
float sum = 0;
for(i = 0; i < n; ++i){
get_next_batch(d, batch, i*batch, X, y);
float err = train_network_datum_gpu(net, X, y);
sum += err;
}
@ -449,7 +467,7 @@ float train_network_sgd(network net, data d, int n)
int i;
float sum = 0;
for(i = 0; i < n; ++i){
get_batch(d, batch, X, y);
get_random_batch(d, batch, X, y);
float err = train_network_datum(net, X, y);
sum += err;
}