From 4223ec79864767b129246be3072b439d1282bc17 Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Tue, 16 Mar 2021 19:18:22 -0700
Subject: [PATCH] fix range

---
 models/Model_SAEHD/Model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py
index 991e805..553e6e8 100644
--- a/models/Model_SAEHD/Model.py
+++ b/models/Model_SAEHD/Model.py
@@ -506,7 +506,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
                             num_labels = self.batch_size
                             for d in tensor.shape[1:]:
                                 num_labels *= d
-                            for _ in num_labels:
+                            for _ in range(num_labels):
                                 if np.random.random() < noise:
                                     label = 0 if label == 1 else 1
                                 if label == 0: