Training test

[sjkoelle]

This adds a unit test for training a single small batch one gradient step, and also makes the landing places of some hyperparameters more explicit.