diff --git a/torch_modules/src/Submodule.cpp b/torch_modules/src/Submodule.cpp
index 24dea4c88955700c8f1a22e92f481da4a15bc099..818a058464318505b42683914085a08876786e79 100644
--- a/torch_modules/src/Submodule.cpp
+++ b/torch_modules/src/Submodule.cpp
@@ -72,6 +72,8 @@ void Submodule::loadPretrainedW2vEmbeddings(torch::nn::Embedding embeddings, std
 
       if (dictIndex >= embeddings->weight.size(0))
       {
+        if ((unsigned long)dictIndex != embeddings->weight.size(0)+toAdd.size())
+          util::myThrow(fmt::format("dictIndex == {}, weight.size == {}, toAdd.size == {}", dictIndex, embeddings->weight.size(0), toAdd.size()));
         toAdd.emplace_back();
         for (unsigned int i = 1; i < splited.size(); i++)
           toAdd.back().emplace_back(std::stof(splited[i]));
@@ -166,7 +168,7 @@ std::function<std::string(const std::string &)> Submodule::getFunction(const std
 
   return [sequence](const std::string & s)
   {
-    auto result = s; 
+    auto result = s;
     for (auto & f : sequence)
       result = f(result);
     return result;