Modify unnecessary syntax

jungyitsai · Jul 24, 2020 · 485344a · 485344a
1 parent 7f59fb0
commit 485344a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -58,7 +58,7 @@ def demo(opt):
                 preds_size = torch.IntTensor([preds.size(1)] * batch_size)
                 _, preds_index = preds.max(2)
                 # preds_index = preds_index.view(-1)
-                preds_str = converter.decode(preds_index.data, preds_size.data)
+                preds_str = converter.decode(preds_index, preds_size)
 
             else:
                 preds = model(image, text_for_pred, is_train=False)