huyluongngoc commited on
Commit
fa5c5c2
·
1 Parent(s): 452dc3a

fix(RCLane): persist GT cache in HF bucket

Browse files
Files changed (1) hide show
  1. 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 \