sort out initialization

[jonathanbratt]

This only matters when training from scratch.
Most layers are currently left with whatever default initialization is already there. Before we do any pre-training, we should make sure that the initialization is set appropriately.