search

HFAiLab / OpenCastKit

The open-source solutions of FourCastNet and GraphCast
MIT License
305 stars 79 forks source link

what the shape of scaler #8

Open liuweishuo opened 1 year ago

liuweishuo commented 1 year ago

https://huggingface.co/hf-ai/OpenCastKit 中下载的scaler.pkl 中channel=21，但是get_data() 处理的数据是维度20，模型也要求是channel维度=20的，这个怎么修改？

VachelHU commented 1 year ago

最后一维度是 total_precipitation , 训练降水模型会用到

zhrli commented 1 year ago

最后一维度是 total_precipitation , 训练降水模型会用到

您的意思是只用前20个， data = (data - scaler['mean']) / scaler['std'] ，infer2img 会报错，所以相应修改？