Skip to content
Snippets Groups Projects
Commit 63d3cf6f authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski
Browse files

Use default tensor constructor.

parent 000157b0
No related merge requests found
Pipeline #2882 passed with stage
in 3 minutes and 15 seconds
...@@ -56,7 +56,8 @@ class TokenCharactersIndexer(token_characters_indexer.TokenCharactersIndexer): ...@@ -56,7 +56,8 @@ class TokenCharactersIndexer(token_characters_indexer.TokenCharactersIndexer):
padded_tokens.pop() padded_tokens.pop()
# Truncates all the tokens to the desired length, and return the result. # Truncates all the tokens to the desired length, and return the result.
return { return {
"token_characters": torch.LongTensor( "token_characters": torch.tensor(
[list(token[:desired_token_length]) for token in padded_tokens] [list(token[:desired_token_length]) for token in padded_tokens],
dtype=torch.long
) )
} }
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment