[🐞] STWave模型test/loss和val/loss为NAN

[weiwei5c5]

Is there an existing issue / discussion for this? | 是否已有关于该错误的issue或讨论？

• I have searched the existing issues / discussions | 我已经搜索过已有的issues和讨论

Is there an existing answer for this in tutorial? | 该问题是否在教程中有解答？

• I have searched tutorial | 我已经搜索过tutorial

Current Behavior | 当前行为

V0.5.8的版本的STWave 跑了pems04 08的数据集

Expected Behavior | 期望行为

希望loss不为nan

Environment | 运行环境

- OS:ubuntu
- DEVICE:linux
- NVIDIA Driver: 575.64.03
- CUDA: 12.9 
- NVIDIA GPU Memory:24576M
- PyTorch: 2.8.0+cu129

BasicTS logs | BasicTS日志

PEMS04.py.txt

training_log_20251127083726.log

cfg.txt

Steps To Reproduce | 复现方法

No response

Anything else? | 备注

No response