polyaxonfile.yml

---
version: 1

kind: experiment

build:
  dockerfile: Dockerfile

run:
  cmd: python train.py

framework: pytorch
environment:
  resources:
    gpu:
      requests: 1
      limits: 1
  node_selector:
    polyaxon: p4-tiny

  persistence:
    data: ['data']
    outputs: 'outputs'

logging:
  level: INFO