Downloads · 30 days
0
djgagne2/fuxi_6h
fuxi_6h is a machine learning model from djgagne2. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
This repository contains the PyTorch checkpoint weights and data/model configuration files for the CREDIT WXFormer 6-hour model. More information about the training and verification of this model can be found in the S…
Downloads · 30 days
0
Access
Public
Updated Jan 24, 2025
Repo size
13.6 GB
Likes
0
Public
Click a slice to open those files.
.pt13.6 GB · 100%
From the Hugging Face model README
This repository contains the PyTorch checkpoint weights and data/model configuration files for the CREDIT WXFormer 6-hour model. More information about the training and verification of this model can be found in the Schreck et al. (2024) ArXiv preprint.
Our model is trained on ERA5 Reanalysis Data from a subset of 16 of the 137 hybrid sigma-pressure vertical levels. The raw data are available on the NSF NCAR Research Data Archive. Processed data can be accessed on the CREDIT ERA5 Zarr Files globus collection.
To run the model, first install the CREDIT package from github.
Modify the paths in the finetune_final/model_predict.yml configuration file to point to the appropriate ERA5 and scaler directories.