jpxkqx commited on
Commit
425dd25
1 Parent(s): 49542f1

Upload 28 files

Browse files
config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ConvSwin2SR"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "depths": [
7
+ 6,
8
+ 6,
9
+ 6,
10
+ 6,
11
+ 6,
12
+ 6
13
+ ],
14
+ "drop_path_rate": 0.1,
15
+ "embed_dim": 180,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "image_size": [
19
+ 32,
20
+ 48
21
+ ],
22
+ "img_range": 1,
23
+ "initializer_range": 0.02,
24
+ "input_shape": [
25
+ 44,
26
+ 60
27
+ ],
28
+ "interpolation_method": "bicubic",
29
+ "layer_norm_eps": 1e-05,
30
+ "mlp_ratio": 2.0,
31
+ "model_type": "conv_swin2sr",
32
+ "num_channels": 1,
33
+ "num_heads": [
34
+ 6,
35
+ 6,
36
+ 6,
37
+ 6,
38
+ 6,
39
+ 6
40
+ ],
41
+ "num_high_res_covars": 0,
42
+ "num_layers": 6,
43
+ "patch_size": 1,
44
+ "qkv_bias": true,
45
+ "real_upscale": 5,
46
+ "resi_connection": "1conv",
47
+ "sample_size": [
48
+ 160,
49
+ 240
50
+ ],
51
+ "torch_dtype": "float32",
52
+ "transformers_version": "4.33.1",
53
+ "upsampler": "pixelshuffle",
54
+ "upscale": 8,
55
+ "use_absolute_embeddings": false,
56
+ "window_size": 5
57
+ }
logs/Train Super-Resolution NN (ConvSwin2SR)/1695130663.064332/events.out.tfevents.1695130663.ebed10aef8f5.321527.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28fd45fc74b8b377f2043752984dc7ab9065bb7c45e0bb2374bb7e1d40105af7
3
+ size 1873
logs/Train Super-Resolution NN (ConvSwin2SR)/1695130663.0680573/hparams.yml ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ batch_size: 2
2
+ device: cuda
3
+ hf_repo_name: predictia/convswin2sr_mediterranean
4
+ hour_embed_type: none
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: domain-wise
10
+ input-variables: t2m
11
+ learning_rate: 0.0001
12
+ lr_warmup_steps: 500
13
+ mixed_precision: fp16
14
+ num_epochs: 100
15
+ num_samples: 3
16
+ num_workers: 0
17
+ number_model_params: 12383377
18
+ output-area: longitude=[-6.85, 5.1],latitude=[44.95, 37]
19
+ output-dataset: cerra
20
+ output-resolution: 005deg
21
+ output-scaling: domain-wise
22
+ output-variable: t2m
23
+ output_dir: /srv/swin2sr-1985_2020
24
+ overwrite_output_dir: true
25
+ push_to_hub: false
26
+ save_image_epochs: 5
27
+ save_model_epochs: 10
28
+ seed: 2023
29
+ static_covariables: null
30
+ train-coverage: 1981/01-2013/12
31
+ validation-coverage: 2014/01-2017/12
logs/Train Super-Resolution NN (ConvSwin2SR)/1695130683.3485534/events.out.tfevents.1695130683.ebed10aef8f5.321527.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf6412ff47016b5116b8bbbc8a36786394858adb6562008f82379fda280345f1
3
+ size 1873
logs/Train Super-Resolution NN (ConvSwin2SR)/1695130683.3505514/hparams.yml ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ batch_size: 2
2
+ device: cuda
3
+ hf_repo_name: predictia/convswin2sr_mediterranean
4
+ hour_embed_type: none
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: domain-wise
10
+ input-variables: t2m
11
+ learning_rate: 0.0001
12
+ lr_warmup_steps: 500
13
+ mixed_precision: fp16
14
+ num_epochs: 100
15
+ num_samples: 3
16
+ num_workers: 0
17
+ number_model_params: 12383377
18
+ output-area: longitude=[-6.85, 5.1],latitude=[44.95, 37]
19
+ output-dataset: cerra
20
+ output-resolution: 005deg
21
+ output-scaling: domain-wise
22
+ output-variable: t2m
23
+ output_dir: /srv/swin2sr-1985_2020
24
+ overwrite_output_dir: true
25
+ push_to_hub: false
26
+ save_image_epochs: 5
27
+ save_model_epochs: 10
28
+ seed: 2023
29
+ static_covariables: null
30
+ train-coverage: 1981/01-2013/12
31
+ validation-coverage: 2014/01-2017/12
logs/Train Super-Resolution NN (ConvSwin2SR)/events.out.tfevents.1695130663.ebed10aef8f5.321527.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d13071628b13c963be8e62d3a11f426096b07e3bf91f0a927c52dd9801dd49d0
3
+ size 88
logs/Train Super-Resolution NN (ConvSwin2SR)/events.out.tfevents.1695130683.ebed10aef8f5.321527.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84842d4dc31c05cf8bcc07d28e75690755affb98ef5ff13822f19e99127131d9
3
+ size 2742875447
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ed5ccc3acc68be004ea0cf728ca9cdf996de62515ffa07142728aeb658d8471
3
+ size 49789685
samples/ConvSwin2SR_0005.png ADDED
samples/ConvSwin2SR_0010.png ADDED
samples/ConvSwin2SR_0015.png ADDED
samples/ConvSwin2SR_0020.png ADDED
samples/ConvSwin2SR_0025.png ADDED
samples/ConvSwin2SR_0030.png ADDED
samples/ConvSwin2SR_0035.png ADDED
samples/ConvSwin2SR_0040.png ADDED
samples/ConvSwin2SR_0045.png ADDED
samples/ConvSwin2SR_0050.png ADDED
samples/ConvSwin2SR_0055.png ADDED
samples/ConvSwin2SR_0060.png ADDED
samples/ConvSwin2SR_0065.png ADDED
samples/ConvSwin2SR_0070.png ADDED
samples/ConvSwin2SR_0075.png ADDED
samples/ConvSwin2SR_0080.png ADDED
samples/ConvSwin2SR_0085.png ADDED
samples/ConvSwin2SR_0090.png ADDED
samples/ConvSwin2SR_0095.png ADDED
samples/ConvSwin2SR_0100.png ADDED