print

2023-01-25 18:06:04 +01:00 · 2023-01-25 18:06:04 +01:00 · 93a231a477
commit 93a231a477
parent ce564bd1c7
1 changed files with 1 additions and 0 deletions
--- a/utils/donut_dataset_stream.py
+++ b/utils/donut_dataset_stream.py
@ -48,6 +48,7 @@ class DonutDataset(Dataset):
        self.added_tokens = added_tokens

        self.dataset = load_dataset(dataset_name_or_path, split=self.split, streaming=True).with_format("torch")
+        print(self.dataset)
        self.dataset_length = len(self.dataset)

        self.gt_token_sequences = []