summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJeff Donahue <jeff.donahue@gmail.com>2015-09-24 21:10:50 (GMT)
committerJeff Donahue <jeff.donahue@gmail.com>2015-09-24 21:10:50 (GMT)
commit674b349522249094b5449f4bc1f90635dc625f4f (patch)
treed5e7833bd2d93fae2c434931a1f2c4dcb7bf2e12
parent349ff65adcab47612dead5142308c2b920797182 (diff)
parent859f93891e4bf47d02899f03f0620fd1f29ca224 (diff)
downloadcaffeonacl-674b349522249094b5449f4bc1f90635dc625f4f.zip
caffeonacl-674b349522249094b5449f4bc1f90635dc625f4f.tar.gz
caffeonacl-674b349522249094b5449f4bc1f90635dc625f4f.tar.bz2
Merge pull request #3115 from lukeyeager/fix-generate-sample-data
Fix generate_sample_data.py
-rw-r--r--src/caffe/test/test_data/generate_sample_data.py4
1 files 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 8349dbb..2645073 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 @@ with h5py.File(script_dir + '/sample_data.h5', 'w') as f:
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 h5py.File(script_dir + '/sample_data_uint8_gzip.h5', 'w') as f:
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