fix train n error
fix train n error

wroscoe/donkey,ricorx7/donkey,alanwells/donkey,ricorx7/donkey,ricorx7/donkey,kennethjiang/donkey,SarthakJShetty/MyDonkey,alanwells/donkey,SarthakJShetty/MyDonkey,kennethjiang/donkey,wroscoe/donkey,ricorx7/donkey,SarthakJShetty/MyDonkey,kennethjiang/donkey,wroscoe/donkey,SarthakJShetty/MyDonkey,alanwells/donkey,wroscoe/donkey