docker-compose.yml

# Compose file to create tensorflow container to export waymo eval data.
# Can use `docker compose build` to create the image, assuming that 
# docker-build.sh was previously run to create the baseline image.
version: '3.3'
services:
  waymo-eval:
    shm_size: 16gb
    tty: true
    build:
      dockerfile: tf.Dockerfile
      args:
        - BASE_IMAGE=$REGISTRY_URL/mp:master
    image: $REGISTRY_URL/mp:tf
    entrypoint:
      # [
      #   python3,
      #   /workspace/evaluate.py,
      #   evaluate,
      #   /ckpt/4c91d8d74a07bb2894158848c74a188b,
      #   --split,
      #   VAL
      # ]
      [
        pytest,
        test/test_waymo_native.py
      ]
    environment:
      - PRETRAINED_ROOT=/ckpt
      - IMAGE_OUT=/workspace
    volumes:
      - type: bind
        source: ./
        target: /workspace
        read_only: false
      - type: bind
        source: /path/to/checkpoints
        target: /ckpt
        read_only: false
      - type: bind
        source: /path/to/datasets
        target: /data
        read_only: false
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [ gpu ]