Continued working on the training function
Continued training.c, finished mu and sigma computations
Working, but dshape is BS.