fixes pvector when bias is none

tfjgeorge · Nov 24, 2023 · 985b443 · 985b443
1 parent 9fb8780
commit 985b443
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nngeometry/object/vector.py b/nngeometry/object/vector.py
@@ -194,7 +194,7 @@ def _dict_to_flat(self):
         parts = []
         for layer_id, layer in self.layer_collection.layers.items():
             parts.append(self.dict_repr[layer_id][0].view(-1))
-            if len(self.dict_repr[layer_id]) > 1:
+            if layer.bias:
                 parts.append(self.dict_repr[layer_id][1].view(-1))
         return torch.cat(parts)