Skip to content
Snippets Groups Projects
Commit 57c43243 authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski Committed by Mateusz Klimaszewski
Browse files

Change tensor constructor.

parent 34dd4929
Branches
Tags
2 merge requests!37Release 1.0.4.,!36Release 1.0.4
...@@ -204,7 +204,7 @@ class UniversalDependenciesDatasetReader(allen_data.DatasetReader): ...@@ -204,7 +204,7 @@ class UniversalDependenciesDatasetReader(allen_data.DatasetReader):
default_value = [0.0] * classes_count default_value = [0.0] * classes_count
padded_tags = util.pad_sequence_to_length(field._indexed_multi_labels, desired_num_tokens, padded_tags = util.pad_sequence_to_length(field._indexed_multi_labels, desired_num_tokens,
lambda: default_value) lambda: default_value)
tensor = torch.LongTensor(padded_tags) tensor = torch.tensor(padded_tags, dtype=torch.long)
return tensor return tensor
return as_tensor return as_tensor
......
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