Spaces:
Configuration error
Configuration error
File size: 1,436 Bytes
a01ef8c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
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
|