jychoi-hpc/ORBIT-2

This readme file was generated on 2025-11-10 by the ORBIT team

GENERAL INFORMATION

1. Title of Dataset: ORBIT-2

2. Author Information: A. Principal Investigator Contact Information

   • Name: Dan Lu
   • ORCID: 0000-0001-5162-9843
   • Institution: ORNL
   • Email: lud1@ornl.gov

   B. Associate or Co-investigator Contact Information

   • Name: Xiao Wang
   • ORCID: 0000-0001-6545-1943
   • Institution: ORNL
   • Email: wangx2@ornl.gov

SHARING/ACCESS INFORMATION

1. Reuse restrictions placed on the data: There are no specific restrictions on the original data, which are publicly available through the links provided above.

2. Links to publications that cite or use the data: The ORBIT-2 publication can be find here: https://arxiv.org/abs/2505.04802.

3. The main ORBIT-2 source is available at: https://github.com/XiaoWang-Github/ORBIT-2

DATA & FILE OVERVIEW

1. File List:

   Two types of model checkpoint files are provided: pretrained and fine-tuned. Pretrained models were trained with three different parameter scales — 117 million, 1 billion, and 8 billion parameters. These models were subsequently fine-tuned on USA regional data, with the resulting checkpoints.

   They are organized into the following folders:

   .
   ├── pretrain
   │   ├── intermediate_117m.ckpt
   │   ├── intermediate_1b.ckpt_rank_0
   │   ├── intermediate_1b.ckpt_rank_1
   │   ├── intermediate_1b.ckpt_rank_2
   │   ├── intermediate_1b.ckpt_rank_3
   │   └── intermediate_8m.ckpt
   └── us-finetune
       ├── 126m_us_precipitation.ckpt
       ├── 9.5m_us_precipitation.ckpt
       ├── temperature_126M.ckpt
       └── temperature_9.5M.ckpt
   

   intermediate_117m.ckpt: Pretrained using 117 million model parameters intermediate_1b.chpt_*: Pretrained using 1 billion model parameters intermediate_8m.chpt: Pretrained using 8 billion model parameters

   126m_us_precipitation.chpt: fine-tuned for US precipitation super-resolution 9.5m_us_precipitation.ckpt: fine-tuned for US precipitation super-resolution temperature_126M.ckpt: fine-tuned for temperature super-resolution temperature_9.5M.ckpt: fine-tuned for temperature super-resolution

   Usage instructions and integration examples for these checkpoint files are available in the main code repository: https://github.com/XiaoWang-Github/ORBIT-2