From fd750423683e87a5b4338b35176b091f169f48e6 Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 10 Jul 2022 16:23:29 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index dd4664d..7679a33 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -24,7 +24,7 @@ loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy']) -model.fit(x_train, y_train, epochs=100) +model.fit(x_train, y_train, epochs=10) model.build() model.summary() sys.exit()