Epoch 5
Browse files- config.json +42 -0
- diffusion_pytorch_model.bin +3 -0
- logs/Train Denoising Diffusion Model/1691156410.3334293/events.out.tfevents.1691156410.5562ac1d6f58.867767.1 +3 -0
- logs/Train Denoising Diffusion Model/1691156410.3378894/hparams.yml +30 -0
- logs/Train Denoising Diffusion Model/events.out.tfevents.1691156410.5562ac1d6f58.867767.0 +3 -0
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
|