Skip to content

Commit 6900045

Browse files
authored
docs(trainer): add note for NumPy randomness (#21310)
1 parent 0bd1993 commit 6900045

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

docs/source-pytorch/common/trainer.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,11 @@ By setting ``workers=True`` in :func:`~lightning.pytorch.seed_everything`, Light
165165
unique seeds across all dataloader workers and processes for :mod:`torch`, :mod:`numpy` and stdlib
166166
:mod:`random` number generators. When turned on, it ensures that e.g. data augmentations are not repeated across workers.
167167

168+
.. note::
169+
170+
If your project depends on NumPy for randomness (e.g. for data
171+
augmentation), it is recommended to use version 2.5.0 or higher.
172+
168173
-------
169174

170175
.. _trainer_flags:

0 commit comments

Comments
 (0)