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

Tmp fixes

Browse Source
This commit is contained in:
FilippoOlivo
2025-03-14 10:20:56 +01:00
parent e780671fd0
commit a3081cc09f
8 changed files with 74 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pina/data/data_module.py
View File

@@ -17,7 +17,7 @@ from ..collector import Collector
class DummyDataloader:
"""
Dataloader used when batch size is `None`. It returns the entire dataset
Dataloader used when batch size is ``None``. It returns the entire dataset
in a single batch.
"""
@@ -38,7 +38,7 @@ class DummyDataloader:
:param dataset: The dataset object to be processed.
:type dataset: PinaDataset
.. note:: This data loader is used when the batch size is `None`.
.. note:: This data loader is used when the batch size is ``None``.
"""
if (
@@ -273,7 +273,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 batch_size: The batch size used for training. If `None`, the
:param batch_size: The batch size used for training. If ``None``, the
entire dataset is returned in a single batch.
:type batch_size: int | None
:param bool shuffle: Whether to shuffle the dataset before splitting.