(#3) Reformer: Hotfix: Cannot use cross entropy

betarixm · Mar 23, 2022 · 4aac5e8 · 4aac5e8
1 parent e2224a4
commit 4aac5e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/defense/models.py b/src/defense/models.py
@@ -16,7 +16,7 @@ def __init__(
         input_shape: tuple,
         name: str = "defense_reformer",
         optimizer: keras.optimizers.Optimizer = keras.optimizers.Adam(),
-        loss: keras.losses.Loss = keras.losses.SparseCategoricalCrossentropy(),
+        loss: keras.losses.Loss = keras.losses.MeanSquaredError(),
         accuracy: keras.metrics.Accuracy = keras.metrics.SparseCategoricalAccuracy(
             name="accuracy"
         ),