From 547aca11bf4e751d1632f2292dbe9b5af3bba569 Mon Sep 17 00:00:00 2001 From: Malte Grosse Date: Thu, 30 Nov 2023 20:36:07 +0900 Subject: [PATCH] 10k 16h --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index 2c93545..ac9da7f 100644 --- a/run.py +++ b/run.py @@ -71,7 +71,7 @@ with tf.device('/GPU:0'): model.compile(optimizer='SGD', loss='categorical_crossentropy', metrics=['accuracy']) - model.fit(X_train_scaled, y_train_categorical, epochs=100) + model.fit(X_train_scaled, y_train_categorical, epochs=10000) model.save('mymodel.keras') print("finished training")