OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed CUDA error checking

This commit is contained in:
AlexeyAB 2019-01-29 13:46:30 +03:00
parent 2790464de1
commit 381f90ebb8
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
build/darknet/x64/cfg/lstm.train.cfg
View File

@ -6,8 +6,9 @@ momentum=0.9
decay=0.001 decay=0.001
max_batches = 2000 max_batches = 2000
time_steps=576 time_steps=576
learning_rate=1.0 learning_rate=0.5
policy=steps policy=steps
burn_in=10
steps=1000,1500 steps=1000,1500
scales=.1,.1 scales=.1,.1

3
cfg/lstm.train.cfg
View File

@ -6,8 +6,9 @@ momentum=0.9
decay=0.001 decay=0.001
max_batches = 2000 max_batches = 2000
time_steps=576 time_steps=576
learning_rate=1.0 learning_rate=0.5
policy=steps policy=steps
burn_in=10
steps=1000,1500 steps=1000,1500
scales=.1,.1 scales=.1,.1

4
src/cuda.c
View File

@ -42,7 +42,7 @@ void check_error(cudaError_t status)
} }
if (status2 != cudaSuccess) if (status2 != cudaSuccess)
{ {
const char *s = cudaGetErrorString(status); const char *s = cudaGetErrorString(status2);
char buffer[256]; char buffer[256];
printf("CUDA Error Prev: %s\n", s); printf("CUDA Error Prev: %s\n", s);
assert(0); assert(0);
@ -153,7 +153,7 @@ void cudnn_check_error(cudnnStatus_t status)
} }
if (status2 != CUDNN_STATUS_SUCCESS) if (status2 != CUDNN_STATUS_SUCCESS)
{ {
const char *s = cudnnGetErrorString(status); const char *s = cudnnGetErrorString(status2);
char buffer[256]; char buffer[256];
printf("cuDNN Error Prev: %s\n", s); printf("cuDNN Error Prev: %s\n", s);
assert(0); assert(0);