Save model using Keras.

README.md

@@ -1,106 +1,17 @@
+
 ---
 library_name: keras
-metrics:
-- TopIQ-FR
-- ArcFace Cosine Distance
-widget:
-- text: input
-  output:
-    url: ./assets/input.png
-- text: target
-  output:
-    url: ./assets/target.png
-- text: output
-  output:
-    url: ./assets/output.png
-pipeline_tag: image-to-image
-base_model:
-- vnet
-- logasja/ArcFace
-tags:
-- adversarial
-- aesthetic
-- quality
-- filter
-datasets:
-- logasja/FDF
-license: gpl-3.0
 ---
-<Gallery />
-
-Training logs [here](https://wandb.ai/spuds/auramask/runs/50d12b8c45c23d741955d72d99b3be51)
 
-# 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 has been uploaded using the Keras library and can be used with JAX,
+TensorFlow, and PyTorch backends.
 
-```json
-{
-    "activation": "ReLU",
-    "batch_norm": false,
-    "filter_num": [
-        64,
-        128,
-        256,
-        512,
-        512
-    ],
-    "n_labels": 3,
-    "output_activation": "tanh",
-    "pool": true,
-    "res_num_ini": 1,
-    "res_num_max": 3,
-    "unpool": false
-}
-```
+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
-{
-    "alpha": 0.0001,
-    "batch": 64,
-    "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.1
-        },
-        "TopIQ": {
-            "full_ref": true,
-            "lower_better": false,
-            "name": "TopIQ",
-            "reduction": "sum_over_batch_size",
-            "score_range": "~0, ~1",
-            "weight": 1
-        }
-    },
-    "mixed_precision": true,
-    "optimizer": {
-        "amsgrad": false,
-        "beta_1": 0.9,
-        "beta_2": 0.999,
-        "clipnorm": 1,
-        "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": "adam",
-        "use_ema": false,
-        "weight_decay": null
-    },
-    "seed": "BIIIIIGSTRETCH",
-    "testing": 0.01,
-    "training": 0.99
-}
-```
+For more details about the model architecture, check out
+[config.json](./config.json).
 
-## Model Architecture Plot
 ![](./assets/summary_plot.png)

metadata.json

@@ -1 +1 @@
-{"keras_version": "3.8.0", "date_saved": "2025-02-17@17:31:49"}
+{"keras_version": "3.8.0", "date_saved": "2025-02-20@15:44:46"}

model.weights.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d8332ee1122867e088d3670ee709201719bcb08d7447cfa795314f216431caa1
-size 145801424
+oid sha256:7509c23c99886a796c52c36b3eac0acbde9c44dd50702c36efd1b21833962f93
+size 137106504