您即将离开知乎，请注意您的账号和财产安全。
https://discuss.pytorch.org/t/nan-loss-with-torch-cuda-amp-and-crossentropyloss/108554/17
