mirror of
https://github.com/velocitatem/PHANTOM.git
synced 2026-05-31 08:33:36 +00:00
feat: tpu orchestrator
This commit is contained in:
8
tpu_orchestration/configs/test_vm.conf
Normal file
8
tpu_orchestration/configs/test_vm.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="us-central2-b"
|
||||
QR_NAME="v4-test-vm"
|
||||
ACCEL_TYPE="v4-8"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv4"
|
||||
IS_SPOT="true"
|
||||
RUN_ID="phantom_v4_test_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
8
tpu_orchestration/configs/v4_od_us.conf
Normal file
8
tpu_orchestration/configs/v4_od_us.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="us-central2-b"
|
||||
QR_NAME="v4-32-us-ondemand"
|
||||
ACCEL_TYPE="v4-32"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv4"
|
||||
IS_SPOT="false"
|
||||
RUN_ID="phantom_v4_od_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
8
tpu_orchestration/configs/v4_spot_us.conf
Normal file
8
tpu_orchestration/configs/v4_spot_us.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="us-central2-b"
|
||||
QR_NAME="v4-32-us-spot"
|
||||
ACCEL_TYPE="v4-32"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv4"
|
||||
IS_SPOT="true"
|
||||
RUN_ID="phantom_v4_spot_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
8
tpu_orchestration/configs/v5e_eu.conf
Normal file
8
tpu_orchestration/configs/v5e_eu.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="europe-west4-b"
|
||||
QR_NAME="v5e-64-eu-spot"
|
||||
ACCEL_TYPE="v5litepod-64"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv5-lite"
|
||||
IS_SPOT="true"
|
||||
RUN_ID="phantom_v5e_eu_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
8
tpu_orchestration/configs/v5e_us.conf
Normal file
8
tpu_orchestration/configs/v5e_us.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="us-central1-a"
|
||||
QR_NAME="v5e-64-us-spot"
|
||||
ACCEL_TYPE="v5litepod-64"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv5-lite"
|
||||
IS_SPOT="true"
|
||||
RUN_ID="phantom_v5e_us_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
8
tpu_orchestration/configs/v6e_eu.conf
Normal file
8
tpu_orchestration/configs/v6e_eu.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="europe-west4-a"
|
||||
QR_NAME="v6e-64-eu-spot"
|
||||
ACCEL_TYPE="v6e-64"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv6e"
|
||||
IS_SPOT="true"
|
||||
RUN_ID="phantom_v6e_eu_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
8
tpu_orchestration/configs/v6e_us.conf
Normal file
8
tpu_orchestration/configs/v6e_us.conf
Normal file
@@ -0,0 +1,8 @@
|
||||
ZONE="us-east1-d"
|
||||
QR_NAME="v6e-64-us-spot"
|
||||
ACCEL_TYPE="v6e-64"
|
||||
RUNTIME_VERSION="v2-alpha-tpuv6e"
|
||||
IS_SPOT="true"
|
||||
RUN_ID="phantom_v6e_us_1"
|
||||
HF_REPO="velocitatem/capstone"
|
||||
TRAIN_CMD="python -m engine.train --sweep-agent --sweep-id lusiana/capstone/oasdorof"
|
||||
Reference in New Issue
Block a user