Skip to content

Commit

Permalink
change tensor to predictions
Browse files Browse the repository at this point in the history
  • Loading branch information
BulatVakhitov committed Jul 16, 2024
1 parent e23a47a commit e3b0428
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion batchflow/models/torch/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -1601,7 +1601,7 @@ def apply_output_operation(predictions, operation):

predictions = predictions[0] if isinstance(predictions, (tuple, list)) else predictions
if operation == 'softplus':
result = torch.nn.functional.softplus(tensor) # pylint: disable=not-callable
result = torch.nn.functional.softplus(predictions) # pylint: disable=not-callable
elif operation == 'sigmoid':
result = torch.sigmoid(predictions)
elif operation == 'sigmoid_uint8':
Expand Down

0 comments on commit e3b0428

Please sign in to comment.