From 473419e7c153d861770e888422f3c8237516ad61 Mon Sep 17 00:00:00 2001 From: Giuseppe Nucifora Date: Wed, 11 Dec 2024 04:19:59 +0100 Subject: [PATCH] add register_keras_serializable --- models/olive_oil/olive_oil-transformer.ipynb | 1 + 1 file changed, 1 insertion(+) diff --git a/models/olive_oil/olive_oil-transformer.ipynb b/models/olive_oil/olive_oil-transformer.ipynb index 734af65..adb8a6b 100644 --- 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",