From 147b8f63f50a0c07713a21dfab9512a48cb26e16 Mon Sep 17 00:00:00 2001 From: Malte Grosse Date: Thu, 30 Nov 2023 10:21:14 +0900 Subject: [PATCH] added --- .woodpecker.yml | 8 ++++++-- run.py | 2 +- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/.woodpecker.yml b/.woodpecker.yml index 9270f82..1a79993 100644 --- a/.woodpecker.yml +++ b/.woodpecker.yml @@ -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 diff --git a/run.py b/run.py index 8f4492d..767117e 100644 --- a/run.py +++ b/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")