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