add grid search

main.py

@@ -667,7 +667,7 @@ if __name__ == "__main__":
                 num_workers=max(0, model.p.num_workers),
                 collate_fn=TrainDataset.collate_fn
             ))
-        for step, batch in dataloader:
+        for step, batch in enumerate(dataloader):
             print(batch[0])
             print(batch.shape)
             sub, rel, obj, label, neg_ent, sub_samp = model.read_batch(