From 7b517fe8c9d0233adc0e96519319b564d7429d38 Mon Sep 17 00:00:00 2001 From: "Ilya V. Schurov" Date: Tue, 7 May 2024 20:45:29 +0200 Subject: [PATCH] Fixing bug in SpinDataset.__iter__ self.shuffle was effectively ignored --- nqs_playground/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nqs_playground/core.py b/nqs_playground/core.py index db434ac..0357d49 100644 --- a/nqs_playground/core.py +++ b/nqs_playground/core.py @@ -309,8 +309,8 @@ def __iter__(self): spins = self.spins values = self.values return zip( - torch.split(self.spins, self.batch_size), - torch.split(self.values, self.batch_size), + torch.split(spins, self.batch_size), + torch.split(values, self.batch_size), )