From 57c432432b1b562a04de3239dccd47ed9ce38754 Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mklimasz@e-science.pl>
Date: Thu, 29 Apr 2021 08:08:18 +0200
Subject: [PATCH] Change tensor constructor.
---
combo/data/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/combo/data/dataset.py b/combo/data/dataset.py
index 870659f..29b1f91 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
--
GitLab