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
|
image: nvcr.io/nvidia/tensorflow:23.10-tf2-py3
|
||||||
commands:
|
commands:
|
||||||
- echo "starting python script"
|
- echo "starting python script"
|
||||||
- zip
|
|
||||||
- python run.py
|
- 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',
|
model.compile(optimizer='SGD',
|
||||||
loss='categorical_crossentropy',
|
loss='categorical_crossentropy',
|
||||||
metrics=['accuracy'])
|
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')
|
model.save('mymodel.keras')
|
||||||
|
|
||||||
print("finished training")
|
print("finished training")
|
||||||
|
|
Loading…
Reference in New Issue