grosse
/
test-ci
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bla
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
Malte Grosse 2024-05-27 10:29:55 +02:00
parent 6e2731ea0f
commit 7a3c6f8393
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
run.py
View File

@ -71,7 +71,7 @@ with tf.device('/GPU:0'):
model.compile(optimizer='SGD', model.compile(optimizer='SGD',
loss='categorical_crossentropy', loss='categorical_crossentropy',
metrics=['accuracy']) metrics=['accuracy'])
model.fit(X_train_scaled, y_train_categorical, epochs=20) model.fit(X_train_scaled, y_train_categorical, epochs=25)
model.save('mymodel.keras') model.save('mymodel.keras')
print("finished training") print("finished training")