Skip to content
Snippets Groups Projects

Correctly pass tensor to device

Merged Yoann Schneider requested to merge correctly-pass-tensor-to-gpu into main
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
+ 1
1
@@ -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)
Loading