peal.generators.ddpm_generator.load_state_dict¶
- peal.generators.ddpm_generator.load_state_dict(path, **kwargs)[source]¶
Load a PyTorch file without redundant fetches across MPI ranks.
- Raises:
ImportError – If the optional
mpi4pydependency is not installed.