Do snapshot after computing loss and test accuracy
Merge pull request #51 from sguada/solver_test

Do snapshot after computing loss and test accuracy