From 03b38c7e99f10830a3d465b6d91e1e4434f6a3e8 Mon Sep 17 00:00:00 2001 From: thanhvc3 Date: Wed, 17 May 2023 13:36:06 +0700 Subject: [PATCH] add grid search --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index bf2f1a3..24c75e5 100644 --- a/main.py +++ b/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(