Improve visualization and allow for a "grey" initialization for convolutional layers.
Change how model initialization is handled, reorganize things a bit in the mode.
Relocate stats file.
Visualize network while training.