aai-institute / continuiti

Learning function operators with neural networks.

GNU Lesser General Public License v3.0

17 stars 3 forks source link

Enhance training output. #110

Closed samuelburbulla closed 3 months ago

samuelburbulla commented 3 months ago

Feature: Enhance training output.

Description

Which issue does this PR tackle?

Enhances the training output.

Epoch 2/2  Step 29/128  [=====               ]   60ms/step  ETA 0:06min - loss/train = 7.7775e-02  loss/test = 8.4045e-02

How does it solve the problem?

Extends the PrintTrainingLoss callback.

How are the changes tested?

No specific test needed.

Checklist for Contributors

[x] Scope: This PR tackles exactly one problem.
[x] Conventions: The branch follows the feature/title-slug convention.
[x] Conventions: The PR title follows the Bugfix: Title convention.
[x] Coding style: The code passes all pre-commit hooks.
[x] Documentation: All changes are well-documented.
[x] Tests: New features are tested and all tests pass successfully.
[x] Changelog: Updated CHANGELOG.md for new features or breaking changes.
[x] Review: A suitable reviewer has been assigned.

Checklist for Reviewers:

[x] The PR solves the issue it claims to solve and only this one.
[x] Changes are tested sufficiently and all tests pass.
[x] Documentation is complete and well-written.
[x] Changelog has been updated, if necessary.