Added test losses to the defaults
Added correct save states function call at the end of the EM iterations
Merge branch 'master' of https://github.com/jeffmahler/abinitio-learning