Add Diffusion example for Kármán vortex street dataset

[frijjoasis]

Modulus Pull Request

Description

Add "A diffusion model for a 2d Kármán vortex street about a fixed cylinder" example

Checklist

• I am familiar with the Contributing Guidelines.
• New or existing tests cover these changes.
• The documentation is up to date with these changes.
• The CHANGELOG.md is up to date with these changes.
• An issue is linked to this pull request.

Dependencies

denoising_diffusion_pytorch
ema_pytorch

[mnabian]

Hi @frijjoasis , thank you for your contribution! I have a few questions about this PR:

• In there a paper corresponding to this model?
• Is this code taken from another repo? If so, could you please provide the link?
• Can the license be changed to Apache 2.0?

[frijjoasis]

Hi, thanks for reviewing.

* In there a paper corresponding to this model?

Yes, it will be uploaded to ArXiv soon.

* Is this code taken from another repo? If so, could you please provide the link?

No, this is the only copy of the code available online. The idea as discussed with one of your team is to link this repo in the paper. The dataset will also be available elsewhere publicly online, and when it is I will update the training script here to download it.

* Can the license be changed to Apache 2.0?

Yes, I'll do that shortly.

[frijjoasis]

Hi, apologies for the delay on this. The paper is now live here: https://arxiv.org/abs/2411.16417

The most recent commit modifies the license and includes a script to download, extract and post-process the dataset from Zenodo https://zenodo.org/records/13820259.