update for codes changes and training logs.