src/network.c
@@ -253,6 +253,7 @@ int i; float *original_input = state.input; float *original_delta = state.delta; state.workspace = net.workspace; for(i = net.n-1; i >= 0; --i){ state.index = i; if(i == 0){ @@ -433,7 +434,7 @@ net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1); #else free(net->workspace); net->workspace = calloc(1, (workspace_size-1)/sizeof(float)+1); net->workspace = calloc(1, workspace_size); #endif //fprintf(stderr, " Done!\n"); return 0;