Skip to content

memmap extension to disk dataset - #1171

Draft
jwa7 wants to merge 1 commit into
mainfrom
mmap_for_diskdataset
Draft

memmap extension to disk dataset#1171
jwa7 wants to merge 1 commit into
mainfrom
mmap_for_diskdataset

Conversation

@jwa7

@jwa7 jwa7 commented Jun 25, 2026

Copy link
Copy Markdown
Member

If specified by the user in options.yaml, writes a temporary memmap dataset from the specified DiskDataset and removes it after training. This keeps the convenience of defining (particularly for electronic structure tagrets) the dataset in the DiskDataset format, but speeds up the IO in the dataloader by using memory mapping.

Contributor (creator of pull-request) checklist

  • Tests updated (for new features and bugfixes)?
  • Documentation updated (for new features)?
  • Issue referenced (for PRs that solve an issue)?

Maintainer/Reviewer checklist

  • CHANGELOG updated with public API or any other important changes?
  • GPU tests passed (maintainer comment: "cscs-ci run")?

📚 Documentation preview 📚: https://metatrain--1171.org.readthedocs.build/en/1171/

Co-authored-by: Copilot <copilot@github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant