Add loss when skipping

GreshamLab · Jul 10, 2024 · 4564bb7 · 4564bb7
1 parent bc9a0ee
commit 4564bb7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/supirfactor_dynamical/training/train_simple_decoders.py b/supirfactor_dynamical/training/train_simple_decoders.py
@@ -112,6 +112,7 @@ def train_simple_multidecoder(
             ):
 
                 if x in skip_decoder_training_models:
+                    _decoder_losses.append(0.)
                     continue
 
                 model_ref.select_submodel(x, 'decoder')