198808xc / Pangu-Weather

An official implementation of Pangu-Weather
1.05k stars 197 forks source link

推理时似乎没有再使用掩码,这是为什么 #66

Open yaleyaleWang opened 1 month ago

yaleyaleWang commented 1 month ago

初学者请教,您在训练时使用了”three constant masks (i.e., the topography mask, land-sea mask and soil type mask) are added to the input of surface variables.“,但是推理时的”Input data preparation using Python“中似乎没用上(”input_surface.npy stores the input surface variables. It is a numpy array shaped (4,721,1440) where the first dimension represents the 4 surface variables (MSLP, U10, V10, T2M in the exact order).“),这是为什么。

198808xc commented 1 month ago

掩码不加在输入数据上,而是融入到了神经网络里面。也就是说,训练和测试过程中,气象变量都是直接作为输入。我们发布的onnx文件是包含了掩码的神经网络,因此直接输入气象变量即可,无需额外处理。

yaleyaleWang commented 1 month ago

thx!