folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
Dario Coscia
2025-03-17 13:47:08 +01:00
committed by FilippoOlivo
parent 9b825845f9
commit f70a23c076
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/data/data_module.py
View File

@@ -278,7 +278,7 @@ class PinaDataModule(LightningDataModule):
be in the range [0, 1].
:param float val_size: Fraction of elements in the validation split. It
must be in the range [0, 1].
:param int batch_size: The batch size used for training. If ``None``,
:param int batch_size: The batch size used for training. If ``None``,
the entire dataset is returned in a single batch.
Default is ``None``.
:param bool shuffle: Whether to shuffle the dataset before splitting.