Upload README.md with huggingface_hub

README.md

@@ -1,17 +1,95 @@
-
 ---
 library_name: keras
 ---
+<Gallery />
+
+Training logs [here](https://wandb.ai/spuds/auramask/runs/5ea0040a6b613278512f5a6c471fc568)
 
-This model has been uploaded using the Keras library and can be used with JAX,
-TensorFlow, and PyTorch backends.
+# Model Description
+This model uses a modified vnet for 2D input/output implemented [here](https://github.com/logasja/keras3-unets) with the following configuration.
 
-This model card has been generated automatically and should be completed by the
-model author.
-See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for
-more information.
+```json
+{
+    "activation": "ReLU",
+    "batch_norm": false,
+    "filter_num": [
+        128,
+        256,
+        512,
+        1024,
+        1024
+    ],
+    "n_labels": 3,
+    "output_activation": "tanh",
+    "pool": false,
+    "res_num_ini": 1,
+    "res_num_max": 3,
+    "unpool": false
+}
+```
 
-For more details about the model architecture, check out
-[config.json](./config.json).
+```json
+{
+    "alpha": 0.0001,
+    "batch": 32,
+    "epochs": 500,
+    "epsilon": 1,
+    "input": "(256, 256)",
+    "losses": {
+        "FEAT_ArcFace": {
+            "d": "cosine_similarity",
+            "f": "ArcFace",
+            "name": "FEAT_ArcFace",
+            "reduction": "sum_over_batch_size",
+            "threshold": 0.68,
+            "weight": 0.05
+        },
+        "FEAT_VGG-Face": {
+            "d": "cosine_similarity",
+            "f": "VGG-Face",
+            "name": "FEAT_VGG-Face",
+            "reduction": "sum_over_batch_size",
+            "threshold": 0.68,
+            "weight": 0.05
+        },
+        "IQASSIMC": {
+            "lower_better": false,
+            "name": "IQASSIMC",
+            "reduction": "sum_over_batch_size",
+            "weight": 0.5
+        },
+        "TopIQ": {
+            "full_ref": true,
+            "lower_better": false,
+            "name": "TopIQ",
+            "reduction": "sum_over_batch_size",
+            "score_range": "~0, ~1",
+            "weight": 0.5
+        }
+    },
+    "mixed_precision": true,
+    "optimizer": {
+        "amsgrad": false,
+        "beta_1": 0.9,
+        "beta_2": 0.999,
+        "clipnorm": null,
+        "clipvalue": null,
+        "ema_momentum": 0.99,
+        "ema_overwrite_frequency": null,
+        "epsilon": 1e-07,
+        "global_clipnorm": null,
+        "gradient_accumulation_steps": null,
+        "learning_rate": 9.999999747378752e-05,
+        "loss_scale_factor": null,
+        "name": "adamw",
+        "use_ema": false,
+        "weight_decay": 0.004
+    },
+    "seed": "BIIIIIGSTRETCH",
+    "testing": 0.01,
+    "training": 0.99
+}
+```
 
+## Model Architecture Plot
 ![](./assets/summary_plot.png)