google-deepmind / graphcast

Apache License 2.0
4.53k stars 572 forks source link

How get the value from model dataset, such as norm_prediction? #70

Open chufall opened 5 months ago

chufall commented 5 months ago

Hi. The code has used the Xarray.dataset to store the intermediate varaiables . Such as , norm_predictions which type is Xarray.dataset composed of several Xarray.DataArray 。

But how get the value from it? When I try to convert it to Numpy.narray with "norm_prediction.values()" , it gave me the following error: {TracerArrayConversionError}The numpy.ndarray conversion method array() was called on traced array with shape float32[1,1,721,1440].

So how fix it? Thanks a lot! qc