From fc1de19bf86b2cfd09264dfc6fbda1937248a40a Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= 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