File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -100,18 +100,18 @@ The webdataset library contains a small wrapper (``WebLoader``) that adds a flui
100
100
import webdataset as wds
101
101
102
102
dataset = wds.WebDataset(
103
- urls,
103
+ urls,
104
104
# needed for multi-gpu or multi-node training
105
105
workersplitter = wds.shardlists.split_by_worker,
106
- nodesplitter = wds.shardlists.split_by_node,
106
+ nodesplitter = wds.shardlists.split_by_node,
107
107
)
108
108
train_dataloader = wds.WebLoader(dataset)
109
109
110
110
model = ...
111
111
trainer = L.Trainer()
112
112
trainer.fit(model, train_dataloader)
113
113
114
- You can find a complete example `here <https://github.com/webdataset/webdataset-lightning >`__.
114
+ You can find a complete example `here <https://github.com/webdataset/webdataset-lightning >`__.
115
115
116
116
NVIDIA DALI
117
117
^^^^^^^^^^^
You can’t perform that action at this time.
0 commit comments