NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems
Apache License 2.0
56 stars 27 forks source link

[other] Add option to load data to host buffers for SFNO #33

Closed daviddpruitt closed 1 year ago

daviddpruitt commented 1 year ago

Description

A large prefetch buffer or validation buffer uses a substantial amount of memory on the GPU. This includes an option to store them on the host.

Info population

Please do not make any changes to the rest of this template. Information will be populated automatically upon submission.

Authors

Authors:

Resources

For technical questions please refer to our forum.

mnabian commented 1 year ago

/blossom-ci

NickGeneva commented 1 year ago

Failed black formatting check

mnabian commented 1 year ago

/blossom-ci