chengtan9907 / OpenSTL

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning
https://openstl.readthedocs.io/en/latest/
Apache License 2.0
730 stars 113 forks source link

自定义nc数据训练问题 #137

Closed StephenHung closed 4 months ago

StephenHung commented 4 months ago

我想利用自己的netCDF数据文件进行训练,但是我把weather数据换成我自己准备的其它nc文件时,训练时的train loss就变成了NaN。请问使用自定义nc数据进行训练时需要怎么做?应该如何修改原始代码才能避免NaN?

StephenHung commented 4 months ago

经调试发现,此问题是nc数据集当中含有缺测值导致的,对xarray dataset对象进行fillna操作后,问题就解决了。