@ -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')
The note is not visible to the blocked user.