aai-institute / continuiti

Learning function operators with neural networks.

GNU Lesser General Public License v3.0

17 stars 3 forks source link

Cleanup: Benchmarks #104

Closed samuelburbulla closed 4 months ago

samuelburbulla commented 4 months ago

Cleanup: Benchmarks

Description

Improve benchmark utility:

Use MLFlow for logging.
Add hyperparameter tuning via optuna.
Improvements in Trainer, Callbacks etc.