changed epoches
moved training with mnist data to own process