diff options
Diffstat (limited to 'src/caffe/layers')
-rw-r--r-- | src/caffe/layers/data_layer.cpp | 2 | ||||
-rw-r--r-- | src/caffe/layers/image_data_layer.cpp | 2 | ||||
-rw-r--r-- | src/caffe/layers/window_data_layer.cpp | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/src/caffe/layers/data_layer.cpp b/src/caffe/layers/data_layer.cpp index a09eb34e..a815c4a6 100644 --- a/src/caffe/layers/data_layer.cpp +++ b/src/caffe/layers/data_layer.cpp @@ -249,7 +249,7 @@ void DataLayer<Dtype>::CreatePrefetchThread() { template <typename Dtype> void DataLayer<Dtype>::JoinPrefetchThread() { - CHECK(!WaitForInternalThreadToExit()) << "Thread joining failed"; + CHECK(WaitForInternalThreadToExit()) << "Thread joining failed"; } template <typename Dtype> diff --git a/src/caffe/layers/image_data_layer.cpp b/src/caffe/layers/image_data_layer.cpp index 8810a381..ce8338bc 100644 --- a/src/caffe/layers/image_data_layer.cpp +++ b/src/caffe/layers/image_data_layer.cpp @@ -172,7 +172,7 @@ void ImageDataLayer<Dtype>::ShuffleImages() { template <typename Dtype> void ImageDataLayer<Dtype>::JoinPrefetchThread() { - CHECK(!WaitForInternalThreadToExit()) << "Thread joining failed"; + CHECK(WaitForInternalThreadToExit()) << "Thread joining failed"; } template <typename Dtype> diff --git a/src/caffe/layers/window_data_layer.cpp b/src/caffe/layers/window_data_layer.cpp index ebd855d4..82bee0ec 100644 --- a/src/caffe/layers/window_data_layer.cpp +++ b/src/caffe/layers/window_data_layer.cpp @@ -413,7 +413,7 @@ void WindowDataLayer<Dtype>::CreatePrefetchThread() { template <typename Dtype> void WindowDataLayer<Dtype>::JoinPrefetchThread() { - CHECK(!WaitForInternalThreadToExit()) << "Thread joining failed."; + CHECK(WaitForInternalThreadToExit()) << "Thread joining failed."; } template <typename Dtype> |