Skip to content

[fix] Set the Linear device equal to the main model device in SoftmaxLoss #25

[fix] Set the Linear device equal to the main model device in SoftmaxLoss

[fix] Set the Linear device equal to the main model device in SoftmaxLoss #25