(shard_id, num_shards, stick_to_shard)
| 106 | def test_caffe_sharding(): |
| 107 | @pipeline_def(batch_size=1, device_id=0, seed=123, num_threads=1) |
| 108 | def pipeline(shard_id, num_shards, stick_to_shard): |
| 109 | images, _ = fn.readers.caffe( |
| 110 | name="Reader", |
| 111 | path=caffe_db_folder, |
| 112 | pad_last_batch=True, |
| 113 | random_shuffle=False, |
| 114 | shard_id=shard_id, |
| 115 | stick_to_shard=stick_to_shard, |
| 116 | num_shards=num_shards, |
| 117 | ) |
| 118 | return images |
| 119 | |
| 120 | def get_data(shard_id, num_shards, stick_to_shard): |
| 121 | p = pipeline(shard_id, num_shards, stick_to_shard) |