Skip to content

Set the Linear device equal to the main model device in SoftmaxLoss (… #29

Set the Linear device equal to the main model device in SoftmaxLoss (…

Set the Linear device equal to the main model device in SoftmaxLoss (… #29