From 8072af59b8a42eb31c5813f025a9744a7a5260c5 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Sun, 7 Nov 2021 17:59:42 +0100
Subject: [PATCH] Concat works with multiple columns
---
common/src/Dict.cpp | 2 +-
torch_modules/src/ContextModule.cpp | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/common/src/Dict.cpp b/common/src/Dict.cpp
index 0eead58..190cd06 100644
--- a/common/src/Dict.cpp
+++ b/common/src/Dict.cpp
@@ -74,7 +74,7 @@ void Dict::readFromFile(const char * filename)
void Dict::insert(const std::string & element)
{
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))
util::myThrow(fmt::format("element '{}' already in dict", element));
diff --git a/torch_modules/src/ContextModule.cpp b/torch_modules/src/ContextModule.cpp
index b1c7fd2..67bbb29 100644
--- a/torch_modules/src/ContextModule.cpp
+++ b/torch_modules/src/ContextModule.cpp
@@ -42,7 +42,7 @@ ContextModuleImpl::ContextModuleImpl(std::string name, const std::string & defin
else if (subModuleType == "GRU")
myModule = register_module("myModule", GRU(columns.size()*inSize, outSize, options));
else if (subModuleType == "Concat")
- myModule = register_module("myModule", Concat(inSize, outSize));
+ myModule = register_module("myModule", Concat(columns.size()*inSize, outSize));
else if (subModuleType == "Transformer")
myModule = register_module("myModule", Transformer(columns.size()*inSize, outSize, options));
else
--
GitLab