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()