[fix
] Set the Linear device equal to the main model device in SoftmaxLoss
#2378
The logs for this run have expired and are no longer available.
Loading
fix
] Set the Linear device equal to the main model device in SoftmaxLoss
#2378