Update m1_tf_test.py
This commit is contained in:
+1
-1
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user