From b81700d223d4670f7cbbb98713d70946b6bab768 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Tue, 25 Feb 2020 20:38:15 +0100
Subject: [PATCH] Added UPOS to the context of RTLSTMNetwork

---
 torch_modules/src/RTLSTMNetwork.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/torch_modules/src/RTLSTMNetwork.cpp b/torch_modules/src/RTLSTMNetwork.cpp
index ef3bfa9..6cc8f70 100644
--- a/torch_modules/src/RTLSTMNetwork.cpp
+++ b/torch_modules/src/RTLSTMNetwork.cpp
@@ -2,12 +2,13 @@
 
 RTLSTMNetworkImpl::RTLSTMNetworkImpl(int nbOutputs, int leftBorder, int rightBorder, int nbStackElements)
 {
-  constexpr int embeddingsSize = 100;
+  constexpr int embeddingsSize = 30;
   constexpr int lstmOutputSize = 500;
   constexpr int hiddenSize = 500;
   setLeftBorder(leftBorder);
   setRightBorder(rightBorder);
   setNbStackElements(nbStackElements);
+  setColumns({"FORM", "UPOS"});
 
   wordEmbeddings = register_module("word_embeddings", torch::nn::Embedding(torch::nn::EmbeddingOptions(50000, embeddingsSize)));
   linear1 = register_module("linear1", torch::nn::Linear(lstmOutputSize, hiddenSize));
-- 
GitLab