Update model.py

src/models/model.py

@@ -42,3 +42,5 @@ class DataModule(Dataset):
         self.target_max_token_len = target_max_token_len
         self.source_max_token_len = source_max_token_len
         self.tokenizer = tokenizer
+
+    def __len__(self):