From fc1de19bf86b2cfd09264dfc6fbda1937248a40a Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Mon, 24 Jun 2024 13:48:01 +0200
Subject: [PATCH] Update.

---
 tasks.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tasks.py b/tasks.py
index 5edb472..64fe967 100755
--- a/tasks.py
+++ b/tasks.py
@@ -154,6 +154,9 @@ class World(Task):
             self.nb_batch_samples_world = input.size(0)
             self.nb_batch_samples_quizzes = 0
 
+        # Shuffle
+        input = input[torch.randperm(input.size(0))]
+
         if desc is None:
             desc = f"epoch-{split}"
         for batch in tqdm.tqdm(
-- 
2.39.5