diff options
| author | Luke Yeager <luke.yeager@gmail.com> | 2015-09-24 13:33:11 -0700 |
|---|---|---|
| committer | Luke Yeager <luke.yeager@gmail.com> | 2015-09-24 13:33:11 -0700 |
| commit | 859f93891e4bf47d02899f03f0620fd1f29ca224 (patch) | |
| tree | d5e7833bd2d93fae2c434931a1f2c4dcb7bf2e12 /src/caffe | |
| parent | 349ff65adcab47612dead5142308c2b920797182 (diff) | |
| download | caffeonacl-859f93891e4bf47d02899f03f0620fd1f29ca224.tar.gz caffeonacl-859f93891e4bf47d02899f03f0620fd1f29ca224.tar.bz2 caffeonacl-859f93891e4bf47d02899f03f0620fd1f29ca224.zip | |
Fix generate_sample_data.py - bug from #2978
Diffstat (limited to 'src/caffe')
| -rw-r--r-- | src/caffe/test/test_data/generate_sample_data.py | 4 |
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 8349dbbc..26450735 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 |