Added counting of epochs run in loss minimizer

Change-Id: I664b37b64c6f2a10a97bd3e8569229971616adb2
2 files changed