Spaces:
Configuration error
Configuration error
Commit ·
fa5c5c2
1
Parent(s): 452dc3a
fix(RCLane): persist GT cache in HF bucket
Browse files- hf_train_carla.sh +2 -0
hf_train_carla.sh
CHANGED
|
@@ -40,6 +40,7 @@ EVAL_WORKERS="${EVAL_WORKERS:-4}" # per GPU
|
|
| 40 |
EVAL_DECODE_WORKERS="${EVAL_DECODE_WORKERS:-18}"
|
| 41 |
CKPT_REPO="${CKPT_REPO:-BanVienCorp/LaneATT-Carla-checkpoints}"
|
| 42 |
CKPT_SUBDIR="${CKPT_SUBDIR:-carla-${VISION}}"
|
|
|
|
| 43 |
EVAL_EVERY="${EVAL_EVERY:-1}"
|
| 44 |
EVAL_SUBSET="${EVAL_SUBSET:---eval-subset 500}"
|
| 45 |
# Resume a previous run in a NEW job: the container is fresh, so first pull last.pth
|
|
@@ -111,6 +112,7 @@ EOF
|
|
| 111 |
hf jobs run \
|
| 112 |
--flavor "${FLAVOR}" \
|
| 113 |
--timeout "${TIMEOUT}" \
|
|
|
|
| 114 |
--secrets HF_TOKEN \
|
| 115 |
--env PYTHONUNBUFFERED=1 \
|
| 116 |
--detach \
|
|
|
|
| 40 |
EVAL_DECODE_WORKERS="${EVAL_DECODE_WORKERS:-18}"
|
| 41 |
CKPT_REPO="${CKPT_REPO:-BanVienCorp/LaneATT-Carla-checkpoints}"
|
| 42 |
CKPT_SUBDIR="${CKPT_SUBDIR:-carla-${VISION}}"
|
| 43 |
+
CACHE_BUCKET="${CACHE_BUCKET:-huyluongngoc/rclane-gt-cache}"
|
| 44 |
EVAL_EVERY="${EVAL_EVERY:-1}"
|
| 45 |
EVAL_SUBSET="${EVAL_SUBSET:---eval-subset 500}"
|
| 46 |
# Resume a previous run in a NEW job: the container is fresh, so first pull last.pth
|
|
|
|
| 112 |
hf jobs run \
|
| 113 |
--flavor "${FLAVOR}" \
|
| 114 |
--timeout "${TIMEOUT}" \
|
| 115 |
+
--volume "hf://buckets/${CACHE_BUCKET}:/workspace/gt_cache:rw" \
|
| 116 |
--secrets HF_TOKEN \
|
| 117 |
--env PYTHONUNBUFFERED=1 \
|
| 118 |
--detach \
|