no mix domain

[GaoN0801]

I wanted to do the cross entropy loss of the source domain and the target domain respectively, and then sum them up，backword total loss. But nan always happens. What's the matter？Due to memory constraints，i set batch_size = 1.