train
- the arguments of Trainer.train now are passed to the fit - unittest for load/restoring from checkpoint