fixed the --step-iter bug, works now

darianyang · Nov 10, 2023 · 3975e4f · 3975e4f
1 parent 441c86e
commit 3975e4f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wedap/h5_pdist.py b/wedap/h5_pdist.py
@@ -285,7 +285,9 @@ def _init_weights(self):
         #weights = [None] * (self.last_iter - weight_start + 1)
         weights = []
         # fill out the weight list
-        for iter in range(weight_start, self.last_iter + 1, self.step_iter):
+        # can't use step-iter here since the entire weight array needs to be indexed by step-iter later
+        #for iter in range(weight_start, self.last_iter + 1, self.step_iter):
+        for iter in range(weight_start, self.last_iter + 1):
             #weights[iter - weight_start] = self.h5[f"iterations/iter_{iter:08d}/seg_index"]["weight"]
             weights.append(self.h5[f"iterations/iter_{iter:08d}/seg_index"]["weight"])
         # 1D array of variably shaped arrays