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

Concat works with multiple columns

parent 0652f587
No related branches found
No related tags found
No related merge requests found
...@@ -74,7 +74,7 @@ void Dict::readFromFile(const char * filename) ...@@ -74,7 +74,7 @@ void Dict::readFromFile(const char * filename)
void Dict::insert(const std::string & element) void Dict::insert(const std::string & element)
{ {
if (element.size() > maxEntrySize) if (element.size() > maxEntrySize)
util::myThrow(fmt::format("inserting element of size={} > maxElementSize={}", element.size(), maxEntrySize)); util::myThrow(fmt::format("inserting element of size={} > maxElementSize={} '{}'", element.size(), maxEntrySize, element));
if (elementsToIndexes.count(element)) if (elementsToIndexes.count(element))
util::myThrow(fmt::format("element '{}' already in dict", element)); util::myThrow(fmt::format("element '{}' already in dict", element));
......
...@@ -42,7 +42,7 @@ ContextModuleImpl::ContextModuleImpl(std::string name, const std::string & defin ...@@ -42,7 +42,7 @@ ContextModuleImpl::ContextModuleImpl(std::string name, const std::string & defin
else if (subModuleType == "GRU") else if (subModuleType == "GRU")
myModule = register_module("myModule", GRU(columns.size()*inSize, outSize, options)); myModule = register_module("myModule", GRU(columns.size()*inSize, outSize, options));
else if (subModuleType == "Concat") else if (subModuleType == "Concat")
myModule = register_module("myModule", Concat(inSize, outSize)); myModule = register_module("myModule", Concat(columns.size()*inSize, outSize));
else if (subModuleType == "Transformer") else if (subModuleType == "Transformer")
myModule = register_module("myModule", Transformer(columns.size()*inSize, outSize, options)); myModule = register_module("myModule", Transformer(columns.size()*inSize, outSize, options));
else else
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment