diff options
| author | Cyprien Noel <cyprien.noel@gmail.com> | 2017-01-04 00:25:00 -0800 |
|---|---|---|
| committer | Cyprien Noel <cyprien.noel@gmail.com> | 2017-01-06 14:57:15 -0800 |
| commit | 5f28eb1147c1abb6e5e5c7cd282218679b0d531d (patch) | |
| tree | 08b8eb88fa0c7b3ea8fe6495c577cf106a4b6fd0 /include | |
| parent | 0d27efc7e3d3d2edbf45cccb73bad03ad655c164 (diff) | |
| download | caffeonacl-5f28eb1147c1abb6e5e5c7cd282218679b0d531d.tar.gz caffeonacl-5f28eb1147c1abb6e5e5c7cd282218679b0d531d.tar.bz2 caffeonacl-5f28eb1147c1abb6e5e5c7cd282218679b0d531d.zip | |
Using default from proto for prefetch
Diffstat (limited to 'include')
| -rw-r--r-- | include/caffe/layers/base_data_layer.hpp | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/include/caffe/layers/base_data_layer.hpp b/include/caffe/layers/base_data_layer.hpp index 925b019d..21d3ada5 100644 --- a/include/caffe/layers/base_data_layer.hpp +++ b/include/caffe/layers/base_data_layer.hpp @@ -67,9 +67,6 @@ class BasePrefetchingDataLayer : virtual void Forward_gpu(const vector<Blob<Dtype>*>& bottom, const vector<Blob<Dtype>*>& top); - // Prefetches batches (asynchronously if to GPU memory) - static const int PREFETCH_COUNT = 4; // same as proto - protected: virtual void InternalThreadEntry(); virtual void load_batch(Batch<Dtype>* batch) = 0; |