File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed
Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments