updated training script
Merge branch 'snapshot-0' into snapshot-2