Execute the output_CIM on the last element of the batch to update the…

… output ports
PrincetonUniversity · Feb 13, 2025 · bd9bb4f · bd9bb4f
1 parent 508e8ef
commit bd9bb4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/psyneulink/library/compositions/pytorchwrappers.py b/psyneulink/library/compositions/pytorchwrappers.py
@@ -814,8 +814,8 @@ def copy_node_values_to_psyneulink(self, nodes:Optional[Union[list,Literal[ALL,
         def update_autodiff_all_output_values():
             """Update autodiff's output_values by executing its output_CIM's with pytorch_rep all_output_values"""
             if self.all_output_values is not None:
-                pass
-                # self._composition.output_CIM.execute(self.all_output_values, context=context)
+                # Execute the output_CIM on the last element of the batch to update the output ports
+                self._composition.output_CIM.execute(self.all_output_values[-1, ...], context=context)
 
         # Allow selective updating of just autodiff.output_values if specified
         if nodes == OUTPUTS: