fix(train): in case of last batch <=2, move to validation if possible

fix(train): in case of last batch <=2, move to validation if possible #2149