Allow passing additional arguments to train.