Update modeling_paragram_sp.py

modeling_paragram_sp.py

@@ -13,11 +13,12 @@ class ParagramSPModel(BertPreTrainedModel):
     def filter_input_ids(self, input_ids):
       output = []
       len = input_ids.shape[1]
-      for ids in input_ids.shape[0]:
+      for i in range(input_ids.shape[0]):
+        ids = input_ids[i]
         filtered_ids = []
-        for i in ids:
-          if i > 0:
-            filtered_ids.append(i)
+        for j in ids:
+          if j > 0:
+            filtered_ids.append(j)
         if len(filtered_ids) == 0:
           filtered_ids = [0]
         output.append(filtered_ids + [config.pad_token_id] * (len - len(filtered_ids)))