add 100 epochs
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
Malte Grosse 2023-11-30 16:29:21 +09:00
parent e55ab32bb1
commit 2701b15f28
1 changed files with 1 additions and 1 deletions

2
run.py
View File

@ -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=10)
model.fit(X_train_scaled, y_train_categorical, epochs=100)
model.save('mymodel.keras')
print("finished training")