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 Nicola Demo
parent 01b29ccbf7
commit 6ac2f28788
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]. be in the range [0, 1].
:param float val_size: Fraction of elements in the validation split. It :param float val_size: Fraction of elements in the validation split. It
must be in the range [0, 1]. 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. the entire dataset is returned in a single batch.
Default is ``None``. Default is ``None``.
:param bool shuffle: Whether to shuffle the dataset before splitting. :param bool shuffle: Whether to shuffle the dataset before splitting.