Trained model

config.json

@@ -0,0 +1,60 @@
+{
+  "architectures": [
+    "ConvSwin2SR"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "conv_channels": [
+    8,
+    8,
+    8
+  ],
+  "conv_kernel_size": [
+    5,
+    5,
+    5
+  ],
+  "conv_padding": [
+    "same",
+    "same",
+    "same"
+  ],
+  "conv_stride": [
+    1,
+    1,
+    1
+  ],
+  "depths": [
+    2,
+    2,
+    2
+  ],
+  "drop_path_rate": 0.1,
+  "embed_dim": 32,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "image_size": [
+    160,
+    240
+  ],
+  "img_range": 1,
+  "initializer_range": 0.02,
+  "layer_norm_eps": 1e-05,
+  "mlp_ratio": 2.0,
+  "model_type": "conv_swin2sr",
+  "num_channels": 1,
+  "num_heads": [
+    2,
+    2,
+    2
+  ],
+  "num_layers": 3,
+  "patch_size": 1,
+  "qkv_bias": true,
+  "resi_connection": "1conv",
+  "torch_dtype": "float32",
+  "transformers_version": "4.30.1",
+  "upsampler": "pixelshuffle",
+  "upscale": 1,
+  "use_absolute_embeddings": false,
+  "window_size": 8
+}

logs/Train Super-Resolution NN/1687270337.4833677/hparams.yml

@@ -0,0 +1,16 @@
+__pydantic_initialised__: true
+batch_size: 16
+device: cuda
+hour_embed_type: none
+hub_private_repo: false
+learning_rate: 0.0001
+lr_warmup_steps: 100
+mixed_precision: fp16
+num_epochs: 100
+num_samples: 3
+output_dir: nn-tiny-subset-domain
+overwrite_output_dir: true
+push_to_hub: false
+save_image_epochs: 5
+save_model_epochs: 5
+seed: 2023

logs/Train Super-Resolution NN/events.out.tfevents.1687270337.wngpu002.18225.0

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

pytorch_model.bin

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