Adds error plot and colourbars to the animation. Start of a loss plot

[cfd1]

Modulus Pull Request

Description

Aims to improve the inference code with an error plot and colourbars. There is a start of the code to plot the loss over the rollout. Needs cleaning up and generalising.

Checklist

• [x] I am familiar with the Contributing Guidelines.
• [ ] New or existing tests cover these changes.
• [ ] The documentation is up to date with these changes.
• [x] The CHANGELOG.md is up to date with these changes.

Dependencies

Depends on PR #111

[cfd1]

The resolution also needs updating as the gifs are quite large now.

[cfd1]

Example of the new visualisation. The model hasn't been trained enough to give good results, but it illustrates the visualisation.

[cfd1]

And an example of the rollout loss plot

[cfd1]

The resolution also needs updating as the gifs are quite large now.

This was fixed in the last commit

[cfd1]

The examples don’t appear to have tests, do I need to add some? For documentation launch only has an ApI doc, is there any documentation for me to update for this change?

[mnabian]

Hi @cfd1 , thanks for the PR! Is this ready for review?

[cfd1]

@mnabian yes, it's ready for a review please

[mnabian]

The examples don’t appear to have tests, do I need to add some? For documentation launch only has an ApI doc, is there any documentation for me to update for this change?

No you don't need to add tests for the examples. For documentation, take a look at the README file for the vortex shedding example and see if anything needs to be updated. Thanks

[mnabian]

@cfd1 thanks again for the PR! It looks good to me, I just left a single comment for you about the resolution.

[mnabian]

/blossom-ci