add register_keras_serializable

2024-12-11 04:19:59 +01:00 · 2024-12-11 04:19:59 +01:00 · 473419e7c1
commit 473419e7c1
parent 32cc73f90f
1 changed files with 1 additions and 0 deletions
--- a/models/olive_oil/olive_oil-transformer.ipynb
+++ b/models/olive_oil/olive_oil-transformer.ipynb
@ -1421,6 +1421,7 @@
    "        decay_steps=7000    # Aumentato per permettere più esplorazione\n",
    "    )\n",
    "\n",
+    "    @keras.saving.register_keras_serializable()\n",
    "    def weighted_huber_loss(y_true, y_pred):\n",
    "        # Pesi per diversi output\n",
    "        weights = tf.constant([1.0, 0.8, 0.8, 1.0, 0.6], dtype=tf.float32)\n",