s444415/donut
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix len

Browse Source
This commit is contained in:
Michał Kozłowski 2023-01-25 21:45:22 +01:00
parent ecce4427a5
commit c4fec90d13
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/donut_dataset_stream.py
View File

@ -116,8 +116,8 @@ class DonutDataset(Dataset):
self.model.decoder.resize_token_embeddings(len(self.processor.tokenizer))
self.added_tokens.extend(list_of_tokens)
# def __len__(self) -> int:
# return self.dataset_length
def __len__(self) -> int:
return self.dataset_length
def __getitem__(self, idx: int) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
"""