28 lines
599 B
Bash
Executable File
28 lines
599 B
Bash
Executable File
#!/bin/bash
|
|
# 从 save_swe_images.sh 生成的 tar.gz 批量加载 swebench 实例镜像。
|
|
|
|
set -e
|
|
|
|
INPUT_DIR="${1:-/data1/sora/temp/evalstone/docker/swe_images}"
|
|
|
|
if [ ! -d "$INPUT_DIR" ]; then
|
|
echo "错误：目录不存在 $INPUT_DIR"
|
|
exit 1
|
|
fi
|
|
|
|
shopt -s nullglob
|
|
TARS=("$INPUT_DIR"/swebench_batch_*.tar.gz)
|
|
TOTAL=${#TARS[@]}
|
|
|
|
echo "共找到 $TOTAL 个镜像包，开始加载..."
|
|
|
|
IDX=0
|
|
for TAR in "${TARS[@]}"; do
|
|
IDX=$((IDX + 1))
|
|
echo "[$IDX/$TOTAL] 加载 $TAR"
|
|
docker load -i "$TAR"
|
|
done
|
|
|
|
echo "完成。已加载镜像数："
|
|
docker images | grep '^swebench/' | wc -l
|