diff options
author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-07-29 18:13:32 -0700 |
---|---|---|
committer | Jeff Donahue <jeff.donahue@gmail.com> | 2014-07-29 18:17:20 -0700 |
commit | afa09ecfa5c09c30824cd0aacf7a7f4e48e70a39 (patch) | |
tree | 1bc9712d9533eecd8dc814a542744b2b75c477bf | |
parent | 6485be076bdebd2ea311e8a06778a55b72a1d095 (diff) | |
download | caffeonacl-afa09ecfa5c09c30824cd0aacf7a7f4e48e70a39.tar.gz caffeonacl-afa09ecfa5c09c30824cd0aacf7a7f4e48e70a39.tar.bz2 caffeonacl-afa09ecfa5c09c30824cd0aacf7a7f4e48e70a39.zip |
Also apply HDF5OutputLayer fix to GPU version.
-rw-r--r-- | src/caffe/layers/hdf5_output_layer.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/hdf5_output_layer.cu b/src/caffe/layers/hdf5_output_layer.cu index 5c790555..9b3dd135 100644 --- a/src/caffe/layers/hdf5_output_layer.cu +++ b/src/caffe/layers/hdf5_output_layer.cu @@ -28,7 +28,7 @@ Dtype HDF5OutputLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, for (int i = 0; i < bottom[0]->num(); ++i) { caffe_copy(data_datum_dim, &bottom[0]->gpu_data()[i * data_datum_dim], &data_blob_.mutable_cpu_data()[i * data_datum_dim]); - caffe_copy(label_datum_dim, &bottom[0]->gpu_data()[i * label_datum_dim], + caffe_copy(label_datum_dim, &bottom[1]->gpu_data()[i * label_datum_dim], &label_blob_.mutable_cpu_data()[i * label_datum_dim]); } SaveBlobs(); |