Skip to content

Commit

Permalink
Update on "Adding lock mechanism to prevent on_disk_cache downloading…
Browse files Browse the repository at this point in the history
… twice"


Fixes #144

Differential Revision: [D36489060](https://our.internmc.facebook.com/intern/diff/D36489060)

[ghstack-poisoned]
  • Loading branch information
VitalyFedyunin committed May 18, 2022
2 parents ae05b84 + 69b3a99 commit 3a9fb53
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions test/test_local_io.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ def init_fn(worker_id):
datapipe = info.dataset
torch.utils.data.graph_settings.apply_sharding(datapipe, num_workers, worker_id)


def _unbatch(x):
return x[0]

Expand Down

0 comments on commit 3a9fb53

Please sign in to comment.