Update README.md

itsubaki · Nov 24, 2024 · cdc7674 · cdc7674
1 parent 5067059
commit cdc7674
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/README.md b/README.md
@@ -118,8 +118,7 @@ for i := 0; i < epochs; i++ {
 	m.ResetState()
 
 	loss, count := variable.New(0), 0
-	for dataloader.Next() {
-		x, t := dataloader.Batch()
+	for x, t := dataloader.Seq2() {
 		y := m.Forward(x)
 		loss = F.Add(loss, F.MeanSquaredError(y, t))