config.yaml

name: training-sem
image: pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
tasks:
  - id: train
    commands:
      - pip install -r requirements.txt
      - python runner/train.py
resources:
  gpu: A100
repository: uav-vts/training_sem