Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 25, 2021

Commit

6718384

1 Parent(s): 5995c80

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -92,7 +92,8 @@ class PLDataModule(LightningDataModule):
             source_max_token_len: int = 512,
             target_max_token_len: int = 512,
             batch_size: int = 4,
-            split: float = 0.1
     ):
         """
         :param data_df:

             source_max_token_len: int = 512,
             target_max_token_len: int = 512,
             batch_size: int = 4,
+            split: float = 0.1,
+            num_workers: int = 2
     ):
         """
         :param data_df: