training

Hello,

Thanks for this amazing work. I just want to train your model on my dataset, but I do not know why all the loss values are zero from the first epoch and did not change at all. Is there anything special to change for training on my dataset?

Thanks in advance.

