diff --git a/combo/data/token_indexers/token_characters_indexer.py b/combo/data/token_indexers/token_characters_indexer.py index ea7a3eaab32a79758573fe6f778c59060d9bce6c..c845fb5d7ef9331fdbbc358a20297e76846f61a3 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 ) }