From 35e46f8352fea35e6107908a709e9708aae32f55 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 24 Mar 2023 12:25:43 +0000
Subject: [PATCH] Correctly pass tensor to device
---
dan/predict/prediction.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index 3cff98d7..d213e8bf 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -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)
--
GitLab