Skip to content
Snippets Groups Projects
Commit b81700d2 authored by Franck Dary's avatar Franck Dary
Browse files

Added UPOS to the context of RTLSTMNetwork

parent b0bb4445
Branches
No related tags found
No related merge requests found
...@@ -2,12 +2,13 @@ ...@@ -2,12 +2,13 @@
RTLSTMNetworkImpl::RTLSTMNetworkImpl(int nbOutputs, int leftBorder, int rightBorder, int nbStackElements) RTLSTMNetworkImpl::RTLSTMNetworkImpl(int nbOutputs, int leftBorder, int rightBorder, int nbStackElements)
{ {
constexpr int embeddingsSize = 100; constexpr int embeddingsSize = 30;
constexpr int lstmOutputSize = 500; constexpr int lstmOutputSize = 500;
constexpr int hiddenSize = 500; constexpr int hiddenSize = 500;
setLeftBorder(leftBorder); setLeftBorder(leftBorder);
setRightBorder(rightBorder); setRightBorder(rightBorder);
setNbStackElements(nbStackElements); setNbStackElements(nbStackElements);
setColumns({"FORM", "UPOS"});
wordEmbeddings = register_module("word_embeddings", torch::nn::Embedding(torch::nn::EmbeddingOptions(50000, embeddingsSize))); wordEmbeddings = register_module("word_embeddings", torch::nn::Embedding(torch::nn::EmbeddingOptions(50000, embeddingsSize)));
linear1 = register_module("linear1", torch::nn::Linear(lstmOutputSize, hiddenSize)); linear1 = register_module("linear1", torch::nn::Linear(lstmOutputSize, hiddenSize));
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment