Skip to content

Commit dc71af4

Browse files
FilippoOlivodario-coscia
authored andcommitted
Minor update
1 parent 9345179 commit dc71af4

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

pina/data/data_module.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -287,14 +287,19 @@ def __init__(
287287
:param automatic_batching: Whether to enable automatic batching.
288288
Default False.
289289
:param int num_workers: Number of worker threads for data loading.
290-
Default 0 (serial loading). For more information, see
291-
https://pytorch.org/docs/stable/data.html#multi-process-data-loading
290+
Default 0 (serial loading).
292291
:param bool pin_memory: Whether to use pinned memory for faster data
293-
transfer to GPU. (Default False). For more information, see
294-
https://pytorch.org/docs/stable/data.html#memory-pinning
292+
transfer to GPU. (Default False).
295293
296294
:raises ValueError: If at least one of the splits is negative.
297295
:raises ValueError: If the sum of the splits is different from 1.
296+
297+
.. seealso::
298+
For more information on multi-process data loading, see:
299+
https://pytorch.org/docs/stable/data.html#multi-process-data-loading
300+
301+
For details on memory pinning, see:
302+
https://pytorch.org/docs/stable/data.html#memory-pinning
298303
"""
299304
super().__init__()
300305

0 commit comments

Comments
 (0)