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:
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(
......
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