change data to dataloader

xrsrke · Oct 8, 2023 · f12afbe · f12afbe
1 parent 6f26709
commit f12afbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -42,10 +42,10 @@ optimizer = torch.optim.Adam(model.parameters())
 + optimizer = DistributedOptimizer(optimizer)
 
 dataset = load_dataset('goose')
-data = torch.utils.data.DataLoader(dataset, shuffle=True)
+dataloader = torch.utils.data.DataLoader(dataset, shuffle=True)
 
 for epoch in range(10):
-    for source, targets in data:
+    for source, targets in dataloader:
 -         source = source.to(device)
 -         targets = targets.to(device)