Added some training steps
There was a little bit of a problem, the face is mosaics, so it might have a little impact,
advised to use 0.8 to 1
