From e3b0428130f135358cdccbe7252f586a592b012f Mon Sep 17 00:00:00 2001 From: BulatVakhitov Date: Tue, 16 Jul 2024 08:15:43 +0000 Subject: [PATCH] change tensor to predictions --- batchflow/models/torch/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py index 50781fd85..929fa4cba 100755 --- a/batchflow/models/torch/base.py +++ b/batchflow/models/torch/base.py @@ -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':