Spaces:

crobbi
/

MNIST-Digit-Classifier

Runtime error

crobbi commited on Sep 3, 2023

Commit

f02d2b7

1 Parent(s): 87723c9

Update modelutil.py

Files changed (1) hide show

modelutil.py CHANGED Viewed

@@ -13,7 +13,7 @@ def create_model():
     LOSS_FUNCTION = tf.keras.losses.SparseCategoricalCrossentropy() # HERE
-    OPTIMIZER = "ADAM"
     METRICS = ["accuracy"]
     model.compile(loss=LOSS_FUNCTION,
     optimizer=OPTIMIZER,

     LOSS_FUNCTION = tf.keras.losses.SparseCategoricalCrossentropy() # HERE
+    OPTIMIZER = tf.keras.optimizers.legacy.Adam()
     METRICS = ["accuracy"]
     model.compile(loss=LOSS_FUNCTION,
     optimizer=OPTIMIZER,