fixing bug in song batch nn training