How to replace the DMC operating device？

[Walhalla-Summary]

I have replaced the DMC running device with CUDA, and the network output is all 0. The output of this action is more like a random selection action. May I ask how to avoid the DMC network output result being 0 when I use CUDA training

[Walhalla-Summary]

tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], device='cuda:0', grad_fn=)

[Walhalla-Summary]

@aetheryang

[daochenzha]

@Walhalla-Summary That is weird. How did you change it to CUDA?

[Walhalla-Summary]

That's not the point. When I use cuda training, all the predicted outputs of the neural network are 0 instead of floating point numbers, which makes me feel strange. Is there any way to solve this problem @daochenzha

[daochenzha]

@Walhalla-Summary How did you change it to CUDA training? More contexts are needed to reproduce the error