From 63d3cf6fb77437c8cb2f0cf5e443d297571ea8ee Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Wed, 14 Apr 2021 12:41:54 +0200
Subject: [PATCH] Use default tensor constructor.
---
combo/data/token_indexers/token_characters_indexer.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/combo/data/token_indexers/token_characters_indexer.py b/combo/data/token_indexers/token_characters_indexer.py
index ea7a3ea..c845fb5 100644
--- a/combo/data/token_indexers/token_characters_indexer.py
+++ b/combo/data/token_indexers/token_characters_indexer.py
@@ -56,7 +56,8 @@ class TokenCharactersIndexer(token_characters_indexer.TokenCharactersIndexer):
padded_tokens.pop()
# Truncates all the tokens to the desired length, and return the result.
return {
- "token_characters": torch.LongTensor(
- [list(token[:desired_token_length]) for token in padded_tokens]
+ "token_characters": torch.tensor(
+ [list(token[:desired_token_length]) for token in padded_tokens],
+ dtype=torch.long
)
}
--
GitLab