Skip to content

Commit a6b4f66

Browse files
authored
Fix Weka refresh (#666)
* Fix Weka refresh * Update evaluate.slurm
1 parent 01b4351 commit a6b4f66

2 files changed

Lines changed: 3 additions & 5 deletions

File tree

slurm/evaluate.slurm

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,9 @@
1212
# Be ye warned this may not work on other clusters!
1313
module load cuda/12.4
1414

15-
# Refresh Weka on h4 cache & current working directory
15+
# Refresh Weka on h4 cache
1616
echo "Refreshing Weka filesystem..."
1717
find -L /fsx/h4/ -type f | xargs -d '\n' -r -n512 -P64 weka fs tier fetch
18-
find -L . -type f -not -path "./data/*" | xargs -d '\n' -r -n512 -P64 weka fs tier fetch
1918

2019
# Needed for vLLM
2120
export VLLM_WORKER_MULTIPROC_METHOD=spawn

slurm/train.slurm

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,9 @@ source openr1/bin/activate
3232
START_TIME=$(date +%s)
3333
echo "START TIME: $(date)"
3434

35-
# Refresh Weka on h4 cache & current working directory
35+
# Refresh Weka on h4 cache
3636
echo "Refreshing Weka filesystem..."
3737
find -L /fsx/h4/ -type f | xargs -d '\n' -r -n512 -P64 weka fs tier fetch
38-
find -L . -type f -not -path "./data/*" | xargs -d '\n' -r -n512 -P64 weka fs tier fetch
3938

4039
# Default values
4140
MODEL=""
@@ -180,4 +179,4 @@ ELAPSED_SECONDS=$((END_TIME - START_TIME))
180179
HOURS=$((ELAPSED_SECONDS / 3600))
181180
MINUTES=$(( (ELAPSED_SECONDS % 3600) / 60 ))
182181
SECONDS=$((ELAPSED_SECONDS % 60))
183-
echo "TOTAL JOB TIME: ${HOURS}h ${MINUTES}m ${SECONDS}s (${ELAPSED_SECONDS} seconds)"
182+
echo "TOTAL JOB TIME: ${HOURS}h ${MINUTES}m ${SECONDS}s (${ELAPSED_SECONDS} seconds)"

0 commit comments

Comments
 (0)