22 lines
448 B
YAML
22 lines
448 B
YAML
matrix:
|
|
EPOCHS:
|
|
- 20
|
|
- 30
|
|
OPTIMIZER:
|
|
- adam
|
|
- SGD
|
|
|
|
steps:
|
|
"train":
|
|
image: nvcr.io/nvidia/tensorflow:23.10-tf2-py3
|
|
commands:
|
|
- echo "starting python script"
|
|
- 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
|
|
|