Skip to content
Snippets Groups Projects
Commit 88d9e114 authored by Marie Generali's avatar Marie Generali :worried:
Browse files

remove unused argument

parent 23eea5d9
No related branches found
No related tags found
No related merge requests found
...@@ -105,7 +105,6 @@ class DAN: ...@@ -105,7 +105,6 @@ class DAN:
start_token=None, start_token=None,
threshold_method="otsu", threshold_method="otsu",
threshold_value=0, threshold_value=0,
temperature=1.0,
): ):
""" """
Run prediction on an input image. Run prediction on an input image.
...@@ -179,7 +178,7 @@ class DAN: ...@@ -179,7 +178,7 @@ class DAN:
num_pred=1, num_pred=1,
) )
pred = pred / temperature pred = pred / self.temperature
whole_output.append(output) whole_output.append(output)
attention_maps.append(weights) attention_maps.append(weights)
confidence_scores.append( confidence_scores.append(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment