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
No related branches found
No related tags found
No related merge requests found
......@@ -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));
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment