add -avg_loss flag