Add layer normalization to the model
Add layer normalization to the model

Add layer normalization to the model
Abstract batch image/label generation into the model

Signed-off-by: Norman Heckscher <de846a94f46d2e80f82b4c588bbdaff401bddcf9@gmail.com>

normanheckscher/mnist-multi-gpu