#347 fixed

nicolay-r · Jun 27, 2022 · 7eafb4c · 7eafb4c
1 parent 0d675fd
commit 7eafb4c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/arekit/contrib/networks/handlers/serializer.py b/arekit/contrib/networks/handlers/serializer.py
@@ -88,7 +88,8 @@ def on_iteration(self, iter_index):
         term_embedding_pairs = collections.OrderedDict()
 
         bpe_vectorizer = BPEVectorizer(embedding=self.__exp_ctx.WordEmbedding, max_part_size=3)
-        norm_vectorizer = RandomNormalVectorizer(vector_size=self.__exp_ctx.WordEmbedding, token_offset=12345)
+        norm_vectorizer = RandomNormalVectorizer(vector_size=self.__exp_ctx.WordEmbedding.VectorSize,
+                                                 token_offset=12345)
 
         text_terms_mapper = StringWithEmbeddingNetworkTermMapping(
             vectorizers={