diff options
author | Sergey Karayev <sergeykarayev@gmail.com> | 2014-07-28 14:55:05 -0700 |
---|---|---|
committer | Sergey Karayev <sergeykarayev@gmail.com> | 2014-07-28 14:55:05 -0700 |
commit | 65c9c4a4c89f9d5112f265d56cc842e47f698afb (patch) | |
tree | 72c154f6b7293cd36bb5c30e0b664dfc73b2ddfb /src/caffe/layers/hdf5_output_layer.cpp | |
parent | cbc507e71fd2615747dd60edf00d7be702aa5aa9 (diff) | |
download | caffeonacl-65c9c4a4c89f9d5112f265d56cc842e47f698afb.tar.gz caffeonacl-65c9c4a4c89f9d5112f265d56cc842e47f698afb.tar.bz2 caffeonacl-65c9c4a4c89f9d5112f265d56cc842e47f698afb.zip |
FIX: both data and label now copied correctly in HDF5 output layer
Diffstat (limited to 'src/caffe/layers/hdf5_output_layer.cpp')
-rw-r--r-- | src/caffe/layers/hdf5_output_layer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/hdf5_output_layer.cpp b/src/caffe/layers/hdf5_output_layer.cpp index accc9f9c..30e49978 100644 --- a/src/caffe/layers/hdf5_output_layer.cpp +++ b/src/caffe/layers/hdf5_output_layer.cpp @@ -55,7 +55,7 @@ Dtype HDF5OutputLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, for (int i = 0; i < bottom[0]->num(); ++i) { caffe_copy(data_datum_dim, &bottom[0]->cpu_data()[i * data_datum_dim], &data_blob_.mutable_cpu_data()[i * data_datum_dim]); - caffe_copy(label_datum_dim, &bottom[0]->cpu_data()[i * label_datum_dim], + caffe_copy(label_datum_dim, &bottom[1]->cpu_data()[i * label_datum_dim], &label_blob_.mutable_cpu_data()[i * label_datum_dim]); } SaveBlobs(); |