Update m1_tf_test.py

This commit is contained in:
Smiril
2022-07-10 16:23:29 +02:00
committed by GitHub
parent 00af86113a
commit fd75042368
+1 -1
View File
@@ -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()