Yes, thank you. After some time I realized that onehot is a probability and to convert it to logit I need to apply a soft version of log(x / (1 - x)) which gave the close to zero results. I’ll changed the title to solved.