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

Minor fixes

Browse Source
This commit is contained in:
AlexeyAB
2019-02-02 03:16:30 +03:00
parent f91d5a5e09
commit 2d747cab2b
5 changed files with 48 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/parser.c
View File

@ -848,7 +848,7 @@ network parse_network_cfg_custom(char *filename, int batch, int time_steps)
free_list(sections);
net.outputs = get_network_output_size(net);
net.output = get_network_output(net);
printf("Total BFLOPS %5.3f \n", bflops);
fprintf(stderr, "Total BFLOPS %5.3f \n", bflops);
#ifdef GPU
get_cuda_stream();
get_cuda_memcpy_stream();
@ -870,7 +870,7 @@ network parse_network_cfg_custom(char *filename, int batch, int time_steps)
check_error(cudaMalloc((void **)net.output16_gpu, *net.max_output16_size * sizeof(short))); //sizeof(half)
}
if (workspace_size) {
printf(" Allocate additional workspace_size = %1.2f MB \n", (float)workspace_size/1000000);
fprintf(stderr, " Allocate additional workspace_size = %1.2f MB \n", (float)workspace_size/1000000);
net.workspace = cuda_make_array(0, workspace_size / sizeof(float) + 1);
}
else {