update training code
bla
bla
bla
bla
bla
bla
bla
bla
bla