From c6ab8d5ad4aaaae86534d37c0eab971c80b9d0a6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A0=BC=E7=89=A9=E8=87=B4=E7=9F=A5?= <532925744@qq.com>
Date: Thu, 25 Aug 2022 10:18:40 +0800
Subject: [PATCH] fixissueI5NVTG
---
official/nlp/lstm/src/imdb.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/official/nlp/lstm/src/imdb.py b/official/nlp/lstm/src/imdb.py
index 7e1976ed7..6fcdd5847 100644
--- a/official/nlp/lstm/src/imdb.py
+++ b/official/nlp/lstm/src/imdb.py
@@ -47,7 +47,7 @@ class ImdbParser():
"""
parse imdb data to memory
"""
- self.__wvmodel = gensim.models.KeyedVectors.load_word2vec_format(self.__glove_file, no_header=True)
+ self.__wvmodel = gensim.models.KeyedVectors.load_word2vec_format(self.__glove_file)
for seg in self.__segs:
self.__parse_imdb_datas(seg)
--
GitLab