Add `build` method to `BaseTrainingPlan` for instatiating `TrainingPlan` from source code or class
Problem: The codes/actions for instantiating TrainingPlan
from class or source code is duplicated in Job
and Round
classes. See Job.__init__
and Round.run_training_round
Solution:
Add BaseTrainingPlan.build
method as classmethod
that will:
-
Take file source, model_arguments
,training_arguments
andoptimizer_args
as argument -
Will load training plan module from class source if source file is provided -
Call post_init
to finalizeTrainingPlan
instantiation.