[fix] Set the Linear device equal to the main model device in SoftmaxLoss#2378
Merged
tomaarsen merged 1 commit intoUKPLab:masterfrom tomaarsen:fix/softmaxloss_deviceDec 13, 2023
+1-1
