try with one-hot labels instead