Fix bug DFIM

ArnovanHilten · Feb 27, 2024 · 682bd45 · 682bd45
1 parent 6baa732
commit 682bd45
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_GenNet.py b/tests/test_GenNet.py
@@ -59,6 +59,8 @@ def __init__(self,
                  suffix='',
                  num_eval = 100,
                  num_sample_pat=100,
+                 start_rank = 0,
+                 end_rank = 0,
                  resultpath = "results/GenNet_experiment_2_/",
                  type = "NID",
                  layer = None):
@@ -84,6 +86,8 @@ def __init__(self,
         self.type = type
         self.layer = layer
         self.num_sample_pat = num_sample_pat
+        self.start_rank = start_rank
+        self.end_rank = end_rank