Skip to content
Snippets Groups Projects
Commit 35e46f83 authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Correctly pass tensor to device

parent 6f97aa8f
No related branches found
No related tags found
1 merge request!85Correctly pass tensor to device
......@@ -100,7 +100,7 @@ class DAN:
:param confidences: Return the characters probabilities.
:param attentions: Return characters attention weights.
"""
input_tensor.to(self.device)
input_tensor = input_tensor.to(self.device)
start_token = len(self.charset) + 1
end_token = len(self.charset)
......
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