version: "3"
services:
  tlt-devel:
    build:
      args:
        http_proxy: ${http_proxy}
        https_proxy: ${https_proxy}
        no_proxy: ""
        IMAGE_NAME: ubuntu
        IMAGE_TAG: 22.04
        PYTHON: python3.10 # Version must be specified for prod
      context: ../
      dockerfile: ./docker/Dockerfile
      target: tlt-devel
    image: intel/ai-tools:tlt-devel-latest
    pull_policy: always
  tlt-prod:
    extends:
      service: tlt-devel
    build:
      args:
        DATASET_DIR: /tmp/data
        OUTPUT_DIR: /tmp/output
      target: tlt-prod
    image: intel/ai-tools:tlt-prod-latest
    volumes:
      - /${DATASET_DIR:-$PWD/../data}:/tmp/data
      - /${OUTPUT_DIR:-$PWD/../output}:/tmp/output
  tlt-dist-devel:
    extends:
      service: tlt-prod
    build:
      args:
        HOROVOD_VERSION: 0.28.0
        ONECCL_VERSION: 2.0.0
        ONECCL_URL: https://developer.intel.com/ipex-whl-stable-cpu
      target: tlt-dist-devel
    image: intel/ai-tools:tlt-dist-devel-latest
  tlt-dist-prod:
    extends:
      service: tlt-dist-devel
    build:
      target: tlt-dist-prod
    command: |
      tlt train -f tensorflow
      --dataset-name cifar10
      --model-name resnet_v1_50
      --dataset-dir /tmp/data
      --output-dir /tmp/output
    environment:
      http_proxy: ${http_proxy}
      https_proxy: ${https_proxy}
      no_proxy: ${no_proxy}
    image: intel/ai-tools:tlt-dist-prod-latest