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(); |