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

This commit is contained in:
Malte Grosse 2023-11-30 10:21:14 +09:00
parent 02ea29fda0
commit 147b8f63f5
2 changed files with 7 additions and 3 deletions

View File

@ -3,7 +3,11 @@ steps:
image: nvcr.io/nvidia/tensorflow:23.10-tf2-py3
commands:
- echo "starting python script"
- zip
- python run.py
"compress and upload":
image: alpine:3
commands:
- apk --no-cache add zip curl
- zip mymodel.zip mymodel.keras
- curl -F fileUpload=@mymodel.zip https://share.storage.sandbox.iuk.hdm-stuttgart.de/upload

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=1)
model.save('mymodel.keras')
print("finished training")