pub(crate) fn run_mnist( mnist_type: MnistType, model_name: Option<String>, batch_size: Option<usize>, learning_rate: Option<f32>, momentum: Option<f32> )