Skip to content

Add options to DataLoader#2138

Merged
clessig merged 2 commits intoecmwf:developfrom
kacpnowak:kacpnowak/develop/dataloader_options
Apr 14, 2026
Merged

Add options to DataLoader#2138
clessig merged 2 commits intoecmwf:developfrom
kacpnowak:kacpnowak/develop/dataloader_options

Conversation

@kacpnowak
Copy link
Copy Markdown
Contributor

Description

Adds option to specify pin_memory and persistent_workers in DataLoader in config file

Issue Number

Closes #2137

Is this PR a draft? Mark it as draft.

Checklist before asking for review

  • I have performed a self-review of my code
  • My changes comply with basic sanity checks:
    • I have fixed formatting issues with ./scripts/actions.sh lint
    • I have run unit tests with ./scripts/actions.sh unit-test
    • I have documented my code and I have updated the docstrings.
    • I have added unit tests, if relevant
  • I have tried my changes with data and code:
    • I have run the integration tests with ./scripts/actions.sh integration-test
    • (bigger changes) I have run a full training and I have written in the comment the run_id(s): launch-slurm.py --time 60
    • (bigger changes and experiments) I have shared a hegdedoc in the github issue with all the configurations and runs for this experiments
  • I have informed and aligned with people impacted by my change:
    • for config changes: the MatterMost channels and/or a design doc
    • for changes of dependencies: the MatterMost software development channel

@github-actions github-actions bot added the model Related to model training or definition (not generic infra) label Mar 30, 2026
@clessig clessig merged commit 1104683 into ecmwf:develop Apr 14, 2026
5 of 6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

model Related to model training or definition (not generic infra)

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

Add option to make workers persistent and pin memory in dataloader

2 participants