Discussion: right location of epoch parameter
We need to discuss if the current implementation, that the
epoch parameter has to be set in the model class, makes sense. Maybe it would be preferable to set this in the experiment up, because this is more related to the experiment than the model. Another reason would be, that if we want to test different epoch numbers, we could trigger this from different run scripts (or even better from a scheduler that distributes different jobs in form of run scripts). Therefore shift
epoch and maybe more parameters to the experiment setup args?