fix bugs

BIT-ENGD · Sep 18, 2021 · 698994a · 698994a
1 parent 8c28a81
commit 698994a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/THUCNews自动摘要.py b/examples/THUCNews自动摘要.py
@@ -26,7 +26,6 @@ def __init__(self) :
         super(BertDataset, self).__init__()
         ## 拿到所有文件名字
         self.txts = glob.glob('./state_dict/THUCNews/*/*.txt')
-
         self.idx2word = {k: v for v, k in word2idx.items()}
         self.tokenizer = Tokenizer(word2idx)
 
@@ -49,7 +48,7 @@ def __getitem__(self, i):
             }
             return output
 
-        self.__getitem__(i + 1)
+        return self.__getitem__(i + 1)
 
     def __len__(self):