• pytorchwrappers.py

- __init__(): modify check for BIAS Nodes
PrincetonUniversity · Jan 3, 2025 · 3de2565 · 3de2565
1 parent 25d119f
commit 3de2565
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/psyneulink/library/compositions/pytorchwrappers.py b/psyneulink/library/compositions/pytorchwrappers.py
@@ -232,8 +232,10 @@ def __init__(self,
                                                        self._composition._get_node_index(node),
                                                        device,
                                                        context=context)
-                pytorch_node._is_bias = all(input_port.default_input == DEFAULT_VARIABLE
-                                            for input_port in node.input_ports)
+                # pytorch_node._is_bias = all(input_port.default_input == DEFAULT_VARIABLE
+                #                             for input_port in node.input_ports)
+                pytorch_node._is_bias = node in self._composition.get_nodes_by_role(NodeRole.BIAS)
+
             self.nodes_map[node] = pytorch_node
             self.wrapped_nodes.append(pytorch_node)