From 859f93891e4bf47d02899f03f0620fd1f29ca224 Mon Sep 17 00:00:00 2001
From: Luke Yeager <luke.yeager@gmail.com>
Date: Thu, 24 Sep 2015 13:33:11 -0700
Subject: [PATCH] Fix generate_sample_data.py - bug from #2978

---
 src/caffe/test/test_data/generate_sample_data.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/caffe/test/test_data/generate_sample_data.py b/src/caffe/test/test_data/generate_sample_data.py
index 8349dbbc8e6..2645073575f 100644
--- a/src/caffe/test/test_data/generate_sample_data.py
+++ b/src/caffe/test/test_data/generate_sample_data.py
@@ -36,7 +36,7 @@
     f['label'] = label
     f['label2'] = label2
 
-with h5py.File(script_dir + '/sample_data_uint8_gzip.h5', 'w') as f:
+with h5py.File(script_dir + '/sample_data_2_gzip.h5', 'w') as f:
     f.create_dataset(
         'data', data=data + total_size,
         compression='gzip', compression_opts=1
@@ -54,7 +54,7 @@
 
 with open(script_dir + '/sample_data_list.txt', 'w') as f:
     f.write('src/caffe/test/test_data/sample_data.h5\n')
-    f.write('src/caffe/test/test_data/sample_uint8_gzip.h5\n')
+    f.write('src/caffe/test/test_data/sample_data_2_gzip.h5\n')
 
 # Generate GradientBasedSolver solver_data.h5