diff options
Diffstat (limited to 'src/caffe/layers/data_layer.cpp')
-rw-r--r-- | src/caffe/layers/data_layer.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/caffe/layers/data_layer.cpp b/src/caffe/layers/data_layer.cpp index fbf2d91c..b57ca587 100644 --- a/src/caffe/layers/data_layer.cpp +++ b/src/caffe/layers/data_layer.cpp @@ -79,6 +79,7 @@ void DataLayer<Dtype>::InternalThreadEntry() { template <typename Dtype> DataLayer<Dtype>::~DataLayer<Dtype>() { + this->JoinPrefetchThread(); // clean up the database resources switch (this->layer_param_.data_param().backend()) { case DataParameter_DB_LEVELDB: |