add grid search

2023-05-17 13:23:12 +07:00 · 2023-05-17 13:23:12 +07:00 · 96b67658f4
commit 96b67658f4
parent 410e725bf2
1 changed files with 1 additions and 0 deletions
--- a/main.py
+++ b/main.py
@ -667,6 +667,7 @@ if __name__ == "__main__":
                collate_fn=TrainDataset.collate_fn
            ))
        for step, batch in dataloader:
+            print(batch.shape)
            sub, rel, obj, label, neg_ent, sub_samp = model.read_batch(
                batch, 'train')