add grid search

2023-05-17 14:08:06 +07:00 · 2023-05-17 14:08:06 +07:00 · c050d8f9d9
commit c050d8f9d9
parent f2052c2839
1 changed files with 1 additions and 1 deletions
--- a/main.py
+++ b/main.py
@ -676,7 +676,7 @@ if __name__ == "__main__":
                neg_ent = neg_ent.cpu()

            inputs = []
-            for i in len(sub):
+            for i in range(len(sub)):
                input = {}
                input['sub'] = sub[i]
                input['rel'] = rel[i]