From 96b67658f44f22c2e21ca9a83ee7188098ad0457 Mon Sep 17 00:00:00 2001 From: thanhvc3 Date: Wed, 17 May 2023 13:23:12 +0700 Subject: [PATCH] add grid search --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index e5fe7f6..ed74963 100644 --- 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')