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