Epoch 10

.gitattributes

@@ -34,3 +34,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 samples/nn_0005.png filter=lfs diff=lfs merge=lfs -text
 samples/nn_0010.png filter=lfs diff=lfs merge=lfs -text
+samples/ConvSwin2SR_0005.png filter=lfs diff=lfs merge=lfs -text
+samples/ConvSwin2SR_0010.png filter=lfs diff=lfs merge=lfs -text

config.json

@@ -0,0 +1,50 @@
+{
+  "architectures": [
+    "ConvSwin2SR"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "depths": [
+    4,
+    4,
+    4
+  ],
+  "drop_path_rate": 0.1,
+  "embed_dim": 32,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "image_size": [
+    32,
+    48
+  ],
+  "img_range": 1,
+  "initializer_range": 0.02,
+  "input_shape": [
+    44,
+    60
+  ],
+  "interpolation_method": "bicubic",
+  "layer_norm_eps": 1e-05,
+  "mlp_ratio": 2.0,
+  "model_type": "conv_swin2sr",
+  "num_channels": 1,
+  "num_heads": [
+    4,
+    4,
+    4
+  ],
+  "num_layers": 3,
+  "patch_size": 1,
+  "qkv_bias": true,
+  "real_upscale": 5,
+  "resi_connection": "1conv",
+  "sample_size": [
+    160,
+    240
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.31.0",
+  "upsampler": "pixelshuffle",
+  "upscale": 8,
+  "use_absolute_embeddings": false,
+  "window_size": 5
+}

logs/Train Super-Resolution NN (ConvSwin2SR)/1690647143.7878377/events.out.tfevents.1690647143.5562ac1d6f58.628257.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c47d1919c593e1ee5f476a34de2c39e554c2f9b10f26be705ca5360bf0ed3bc
+size 1163

logs/Train Super-Resolution NN (ConvSwin2SR)/1690647143.7918205/hparams.yml

@@ -0,0 +1,17 @@
+batch_size: 2
+device: cuda
+hf_repo_name: predictia/europe_reanalysis_downscaler_convswin2sr
+hour_embed_type: none
+hub_private_repo: false
+learning_rate: 0.001
+lr_warmup_steps: 500
+mixed_precision: fp16
+num_epochs: 200
+num_samples: 3
+number_model_params: 649849
+output_dir: swin2sr-1985_2020
+overwrite_output_dir: true
+push_to_hub: true
+save_image_epochs: 5
+save_model_epochs: 10
+seed: 2023

logs/Train Super-Resolution NN (ConvSwin2SR)/events.out.tfevents.1690647143.5562ac1d6f58.628257.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e86fa55b0f7bbef5c6b166c06ff78c563cf9484b1711572e72b714ab5c999d3
+size 123935918

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae0a4fade900f59bc5180bf181cfa7ed89be8e3c86525615e1a0614e2798475a
+size 2692901