Adds sampling and reconstruction error
Writes train and validation cross entropy