Hugging Face's logo

easylearning
/
Triton_Earth_V1

Model card Files
xet
Community
Triton_Earth_V1
File size: 855 Bytes
408f9e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
 



## 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
```