fixing bug in nn song batch size