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

add epochs to py
ci/woodpecker/push/woodpecker/1 Pipeline was successful Details
ci/woodpecker/push/woodpecker/2 Pipeline was successful Details

This commit is contained in:
Cornelius Specht 2024-06-13 11:28:56 +02:00
parent 05291f41f6
commit e24b6eef5a
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.woodpecker.yml
View File

@ -1,5 +1,5 @@
matrix:
MSTEPS:
EPOCHS:
- 20
- 30

6
run.py
View File

@ -8,6 +8,10 @@ import os
from keras import backend as K
K.clear_session()
EPOCHS = int(os.getenv("EPOCHS", default = 10))
gpus = tf.config.experimental.list_physical_devices('GPU')
# if gpus:
# try:
@ -71,7 +75,7 @@ with tf.device('/GPU:0'):
model.compile(optimizer='SGD',
loss='categorical_crossentropy',
metrics=['accuracy'])
model.fit(X_train_scaled, y_train_categorical, epochs=25)
model.fit(X_train_scaled, y_train_categorical, epochs=EPOCHS)
model.save('mymodel.keras')
print("finished training")