## Installation

- cuda 11.8

```
conda env create -f environment.yml
conda activate triton
```

## Notes

The full project is avilable on [Hugging Face](https://huggingface.co/easylearning/Triton_Earth_V1/tree/main), you can find the pretrained model, test data on Hugging Face and put them in the same location.

## Data Structure

Preparing the train, valid, and test data as follows:

```
./data/
|--train
|  |--1993.h5
|  |--1994.h5
|  |--......
|  |--2016.h5
|  |--2017.h5
|--valid
|  |--2018.h5
|  |--2019.h5
|--test
|  |--2020.h5
|--mean_s_t_ssh.npy
|--std_s_t_ssh.npy
|--climate_mean_s_t_ssh.npy
|--land_mask.h5
```

## Inference

```
sh inference.sh
```
   
## Training

- **Single-node Multi-GPU Training**

```
sh train_single_node_and_multi_gpus.sh
```

- **Multi-node Multi-GPU Training**

```
sh train_multi_nodes_and_multi_gpus.sh
```