cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add grid search

Browse Source
This commit is contained in:
thanhvc3 2023-05-17 13:36:06 +07:00
parent 87f45862d6
commit 03b38c7e99
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

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