Skip to content

Commit

Permalink
Do not log validation worker seed initialization
Browse files Browse the repository at this point in the history
  • Loading branch information
mittagessen committed Apr 30, 2024
1 parent c621747 commit ffe6de3
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions kraken/lib/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,10 +67,7 @@ def _validation_worker_init_fn(worker_id):
for lightning because otherwise it will display a message
at info level about the seed being changed. """
from lightning.pytorch import seed_everything
level = logging.getLogger("lightning_fabric.utilities.seed").level
logging.getLogger("lightning_fabric.utilities.seed").setLevel(logging.WARN)
seed_everything(42)
logging.getLogger("lightning_fabric.utilities.seed").setLevel(level)


class KrakenTrainer(L.Trainer):
Expand Down

0 comments on commit ffe6de3

Please sign in to comment.