From 381f90ebb8d1518141e809521fe2985a2d506fc1 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Tue, 29 Jan 2019 13:46:30 +0300 Subject: [PATCH] Fixed CUDA error checking --- build/darknet/x64/cfg/lstm.train.cfg | 3 ++- cfg/lstm.train.cfg | 3 ++- src/cuda.c | 4 ++-- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/build/darknet/x64/cfg/lstm.train.cfg b/build/darknet/x64/cfg/lstm.train.cfg index 874b990e..f0d4721d 100644 --- a/build/darknet/x64/cfg/lstm.train.cfg +++ b/build/darknet/x64/cfg/lstm.train.cfg @@ -6,8 +6,9 @@ momentum=0.9 decay=0.001 max_batches = 2000 time_steps=576 -learning_rate=1.0 +learning_rate=0.5 policy=steps +burn_in=10 steps=1000,1500 scales=.1,.1 diff --git a/cfg/lstm.train.cfg b/cfg/lstm.train.cfg index 874b990e..f0d4721d 100644 --- a/cfg/lstm.train.cfg +++ b/cfg/lstm.train.cfg @@ -6,8 +6,9 @@ momentum=0.9 decay=0.001 max_batches = 2000 time_steps=576 -learning_rate=1.0 +learning_rate=0.5 policy=steps +burn_in=10 steps=1000,1500 scales=.1,.1 diff --git a/src/cuda.c b/src/cuda.c index 16af413f..77f4dd71 100644 --- a/src/cuda.c +++ b/src/cuda.c @@ -42,7 +42,7 @@ void check_error(cudaError_t status) } if (status2 != cudaSuccess) { - const char *s = cudaGetErrorString(status); + const char *s = cudaGetErrorString(status2); char buffer[256]; printf("CUDA Error Prev: %s\n", s); assert(0); @@ -153,7 +153,7 @@ void cudnn_check_error(cudnnStatus_t status) } if (status2 != CUDNN_STATUS_SUCCESS) { - const char *s = cudnnGetErrorString(status); + const char *s = cudnnGetErrorString(status2); char buffer[256]; printf("cuDNN Error Prev: %s\n", s); assert(0);