okeowo1014 commited on
Commit
4a762d5
·
1 Parent(s): 07d6fbd

up date trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +2 -1
trainer.py CHANGED
@@ -73,4 +73,5 @@ model.fit(train_images, train_labels, epochs=10, validation_split=0.1)
73
  model.evaluate(test_images, test_labels)
74
 
75
  # save the model
76
- model.save("model.h5")
 
 
73
  model.evaluate(test_images, test_labels)
74
 
75
  # save the model
76
+ model.save("model.keras")
77
+