aai-institute / continuiti

Learning function operators with neural networks.
GNU Lesser General Public License v3.0
21 stars 3 forks source link

Cleanup: DataLoader in Trainer #65

Closed samuelburbulla closed 7 months ago

samuelburbulla commented 7 months ago

Cleanup: DataLoader in Trainer

Description

As everything related to training moved into the Trainer class, it is consequent to also move the DataLoader to the 'fit' methods as it, particularly, can take care of parallel training if available.

Which issue does this PR tackle?

How does it solve the problem?

How are the changes tested?

Checklist for Contributors

Checklist for Reviewers: