From 88d9e11444e5a81fac7d6cc84e9ed1f3826ebcf0 Mon Sep 17 00:00:00 2001
From: M Generali <mgenerali@teklia.com>
Date: Tue, 6 Jun 2023 17:23:31 +0200
Subject: [PATCH] remove unused argument
---
dan/predict/prediction.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index d3e7a5fe..8d021712 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -105,7 +105,6 @@ class DAN:
start_token=None,
threshold_method="otsu",
threshold_value=0,
- temperature=1.0,
):
"""
Run prediction on an input image.
@@ -179,7 +178,7 @@ class DAN:
num_pred=1,
)
- pred = pred / temperature
+ pred = pred / self.temperature
whole_output.append(output)
attention_maps.append(weights)
confidence_scores.append(
--
GitLab