Skip to content

Commit

Permalink
-
Browse files Browse the repository at this point in the history
  • Loading branch information
jdcpni committed Nov 22, 2024
1 parent ae4e8d8 commit 99cd2ec
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion psyneulink/library/compositions/autodiffcomposition.py
Original file line number Diff line number Diff line change
Expand Up @@ -1063,7 +1063,6 @@ def _get_loss(self, loss_spec):
# and therefore requires a wrapper function to properly package inputs.
return lambda x, y: nn.CrossEntropyLoss()(torch.atleast_2d(x), torch.atleast_2d(y.type(x.type())))
elif loss_spec == Loss.BINARY_CROSS_ENTROPY:
if version.parse(torch.version.__version__) >= version.parse('1.12.0'):
return nn.BCELoss()
elif loss_spec == Loss.L1:
return nn.L1Loss(reduction='sum')
Expand Down

0 comments on commit 99cd2ec

Please sign in to comment.