diff --git a/combo/data/dataset.py b/combo/data/dataset.py index 870659fe27857d157e5061f77690f61f166330a7..29b1f918c5de55f8055ebb432d3fc90e97ba2c42 100644 --- a/combo/data/dataset.py +++ b/combo/data/dataset.py @@ -204,7 +204,7 @@ class UniversalDependenciesDatasetReader(allen_data.DatasetReader): default_value = [0.0] * classes_count padded_tags = util.pad_sequence_to_length(field._indexed_multi_labels, desired_num_tokens, lambda: default_value) - tensor = torch.LongTensor(padded_tags) + tensor = torch.tensor(padded_tags, dtype=torch.long) return tensor return as_tensor