added
ci/woodpecker/push/woodpecker Pipeline was successful
Details
ci/woodpecker/push/woodpecker Pipeline was successful
Details
This commit is contained in:
parent
02ea29fda0
commit
147b8f63f5
|
@ -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
2
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=10)
|
||||
model.fit(X_train_scaled, y_train_categorical, epochs=1)
|
||||
model.save('mymodel.keras')
|
||||
|
||||
print("finished training")
|
||||
|
|
Loading…
Reference in New Issue