| 2 | |
| 3 | |
| 4 | def make_data_reader(lbann): |
| 5 | reader = lbann.reader_pb2.DataReader() |
| 6 | |
| 7 | # Test data reader |
| 8 | _reader = reader.reader.add() |
| 9 | _reader.name = "python" |
| 10 | _reader.role = "test" |
| 11 | _reader.shuffle = False |
| 12 | _reader.fraction_of_data_to_use = 1 |
| 13 | _reader.python.module = "iur_dataset" |
| 14 | _reader.python.module_dir = os.path.dirname(os.path.realpath(__file__)) |
| 15 | _reader.python.sample_function = "get_test_sample" |
| 16 | _reader.python.num_samples_function = "num_test_samples" |
| 17 | _reader.python.sample_dims_function = "sample_dims" |
| 18 | |
| 19 | return reader |