Skip to content

Commit

Permalink
tchaton patch 1 (#357)
Browse files Browse the repository at this point in the history
* Update README.md

* update

* update
  • Loading branch information
tchaton authored Sep 3, 2024
1 parent 9f55eef commit 111c2fe
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,11 @@ Load the data by replacing the PyTorch DataSet and DataLoader with the Streaming
```python
import litdata as ld

train_dataset = ld.StreamingDataset('s3://my-bucket/my_optimized_dataset', shuffle=True, drop_last=True)
train_dataset = ld.StreamingDataset(
's3://my-bucket/my_optimized_dataset',
shuffle=True,
drop_last=True,
)
train_dataloader = ld.StreamingDataLoader(train_dataset)

for sample in train_dataloader:
Expand Down

0 comments on commit 111c2fe

Please sign in to comment.