glxe
glxe
默认参数没有变，数据的话是我自己的数据。但是也都是处理你的这个格式的。 用的v100的卡，第一次发现这个情况的环境是 cuda 12.1 , torch 应该是1.7.1。 然后我把cuda 降到11.6 torch 1.7.1，问题依旧。 然后cuda又降到11.2（不过这个版本没跑起来，可能环境问题）。
目前check的数据没看出有啥问题，预测也试了，但是实际推理的结果，很离谱，同样的测试集，用其他模型推理，结果在预期内，用这个训练好的模型，结果就是飞的很离谱，就很奇怪。
好像是环境问题，我又重新折腾下了系统环境和驱动，就好了
NVIDIA-SMI 460.32.03 Driver Version: 460.32.03 CUDA Version: 11.2 torch 1.7.1 最后就是装的这个驱动版本，然后就好了。 你可以试试这个环境