fix
This commit is contained in:
parent
28d1edb830
commit
2072b012af
|
@ -76,9 +76,7 @@ def main():
|
||||||
epochs=config.EPOCHS,
|
epochs=config.EPOCHS,
|
||||||
# steps_per_epoch=math.ceil(len(train_dataset) / config.BATCH_SIZE),
|
# steps_per_epoch=math.ceil(len(train_dataset) / config.BATCH_SIZE),
|
||||||
callbacks = [early_stopping],
|
callbacks = [early_stopping],
|
||||||
)s=2
|
)
|
||||||
f=9
|
|
||||||
s+f
|
|
||||||
|
|
||||||
|
|
||||||
# plt.plot(history.history['loss'])
|
# plt.plot(history.history['loss'])
|
||||||
|
@ -117,4 +115,6 @@ def main():
|
||||||
model.save("captcha_solver.keras")
|
model.save("captcha_solver.keras")
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
# main()
|
||||||
|
model = keras.saving.load_model("test.keras")
|
||||||
|
model.summary()
|
Loading…
Reference in New Issue