Proposal: Pluggable `ModelTrainer` train function#3084

Merged

alanakbik merged 15 commits intoflairNLP:pluggable_trainerfrom

plonerma:pluggable_trainer

Mar 24, 2023

Commits on Feb 21, 2023

Minor improvements in training behavior plugin
plonerma
committed

Commits on Feb 22, 2023

Commits on Feb 23, 2023

Pass epoch to collecting_train_return_values hook (may be relevant for metric-recording)
plonerma
committed
Count total batches in plugin instead of calculating it (when training a second time, this is more flexibel)
plonerma
committed

Commits on Mar 16, 2023

Changed collection of return values
plonerma
committed

Commits on Mar 21, 2023

Commits on Mar 22, 2023

Removed empty newline
plonerma
committed