| 97 | |
| 98 | template <typename Dtype> |
| 99 | void MemoryDataLayer<Dtype>::set_batch_size(int new_size) { |
| 100 | CHECK(!has_new_data_) << |
| 101 | "Can't change batch_size until current data has been consumed."; |
| 102 | batch_size_ = new_size; |
| 103 | added_data_.Reshape(batch_size_, channels_, height_, width_); |
| 104 | added_label_.Reshape(batch_size_, 1, 1, 1); |
| 105 | } |
| 106 | |
| 107 | template <typename Dtype> |
| 108 | void MemoryDataLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, |