jpxkqx commited on
Commit
ea2c265
1 Parent(s): 029d841
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 16,
9
+ 24,
10
+ 32
11
+ ],
12
+ "center_input_sample": false,
13
+ "class_embed_type": null,
14
+ "down_block_types": [
15
+ "DownBlock2D",
16
+ "AttnDownBlock2D",
17
+ "AttnDownBlock2D"
18
+ ],
19
+ "downsample_padding": 1,
20
+ "downsample_type": "conv",
21
+ "flip_sin_to_cos": true,
22
+ "freq_shift": 0,
23
+ "in_channels": 2,
24
+ "layers_per_block": 3,
25
+ "mid_block_scale_factor": 1,
26
+ "norm_eps": 1e-05,
27
+ "norm_num_groups": 4,
28
+ "num_class_embeds": 24,
29
+ "out_channels": 1,
30
+ "resnet_time_scale_shift": "default",
31
+ "sample_size": [
32
+ 160,
33
+ 240
34
+ ],
35
+ "time_embedding_type": "positional",
36
+ "up_block_types": [
37
+ "AttnUpBlock2D",
38
+ "AttnUpBlock2D",
39
+ "UpBlock2D"
40
+ ],
41
+ "upsample_type": "conv"
42
+ }
diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a042b26461cec54ac240ed215d5ba90a41c94ca722b7de6c3450d29f52b0d71e
3
+ size 2041915
logs/Train Denoising Diffusion Model/1691156410.3334293/events.out.tfevents.1691156410.5562ac1d6f58.867767.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81a2a2a4ee241b00300853427f0109c151a1c84419fc53b6af0f04890423b5e3
3
+ size 1837
logs/Train Denoising Diffusion Model/1691156410.3378894/hparams.yml ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ batch_size: 4
2
+ device: cuda
3
+ hf_repo_name: predictia/cerra_denoise_nn_pixel-scaled_large
4
+ hour_embed_type: class
5
+ hub_private_repo: false
6
+ input-area: longitude=[-8.35, 6.6],latitude=[46.45, 35.5]
7
+ input-dataset: era5
8
+ input-resolution: 025deg
9
+ input-scaling: pixel-wise
10
+ input-variables: t2m
11
+ learning_rate: 0.001
12
+ lr_warmup_steps: 500
13
+ mixed_precision: fp16
14
+ num_epochs: 30
15
+ num_samples: 3
16
+ number_model_params: 470129
17
+ output-area: longitude=[-6.85, 5.1],latitude=[44.95, 37]
18
+ output-dataset: cerra
19
+ output-resolution: 005deg
20
+ output-scaling: pixel-wise
21
+ output-variable: t2m
22
+ output_dir: cddpm-probando-large
23
+ overwrite_output_dir: true
24
+ push_to_hub: true
25
+ save_image_epochs: 10
26
+ save_model_epochs: 5
27
+ seed: 2023
28
+ static_covariables: null
29
+ train-coverage: 1981/01-2013/12
30
+ validation-coverage: 2014/01-2017/12
logs/Train Denoising Diffusion Model/events.out.tfevents.1691156410.5562ac1d6f58.867767.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9ecba7add862d08d5f4290d62ebd74e27079ef1fb6cc9f35e3d3f2a5bee96eb
3
+ size 33584671