Repository navigation
Expand file tree
/
Copy pathconfig.env
More file actions
39 lines (33 loc) · 1.64 KB
/
Copy pathconfig.env
File metadata and controls
39 lines (33 loc) · 1.64 KB
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
# =============================================================================
# Benchmark-Lab Configuration
# =============================================================================
# This file is sourced by run_experiments.sh and my_test.sh.
# Edit values here instead of passing CLI flags every time.
# CLI flags (e.g. --bucket) still override these defaults.
# =============================================================================
# ---- Run Configuration ----
# "local" = runs on this machine via Docker Compose
# "batch" = submits job to AWS Batch
EXPERIMENT_MODE="batch"
# ---- AWS General ----
AWS_REGION="us-east-2"
S3_BUCKET="experiment-sandbox"
# S3 key prefix under which results are stored (run key is appended)
S3_PREFIX="benchmark-lab-test"
# ---- AWS ECR ----
# ECR repository URI for pushing the experiment image (batch mode only)
# e.g. 123456789012.dkr.ecr.us-east-2.amazonaws.com/benchmark-lab
ECR_REPO="411614151856.dkr.ecr.us-east-2.amazonaws.com/experiments/benchmark-lab-test"
# ---- AWS Batch ----
# These fields are ONLY used when EXPERIMENT_MODE="batch"
BATCH_JOB_DEFINITION="experiment-general-definition:5"
BATCH_JOB_QUEUE="experiments-sandbox-ec2"
# Container-level resource overrides (sent with each job submission).
# Set to "" to use the defaults baked into the job definition.
# ONLY used when EXPERIMENT_MODE="batch"
BATCH_VCPUS="4" # Number of vCPUs for the container
BATCH_MEMORY="16384" # Memory in MiB (16384 = 16 GB)
BATCH_GPUS="" # Number of GPUs (leave empty for none)
# ---- Local / Docker ----
# Extra flags passed to `docker compose up` (e.g. "--no-build")
DOCKER_COMPOSE_EXTRA_FLAGS=""