add train to loss monitoring