Fix mask layering

GreshamLab · Oct 17, 2023 · 5714cd9 · 5714cd9
1 parent fea1c25
commit 5714cd9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/supirfactor_dynamical/models/chromatin_model.py b/supirfactor_dynamical/models/chromatin_model.py
@@ -54,7 +54,7 @@ def __init__(
 
         self.mask_input_weights(
             gene_peak_mask,
-            module=self.peak_encoder[1],
+            module=self.peak_encoder[0],
             layer_name='weight'
         )
 
@@ -70,7 +70,7 @@ def __init__(
 
         self.mask_input_weights(
             peak_tf_mask,
-            module=self.tf_encoder[0],
+            module=self.tf_encoder[1],
             layer_name='weight'
         )