gaojie
c66855adfc
Sync to site1 / sync (push) Has been cancelled
- 移除 JEPA/lejepa-identifiability 子模块 gitlink - 移除 research/multiply/MultiPLY 子模块 gitlink - 删除 .gitmodules（不再有外部 URL 依赖） - 两个目录内容作为普通文件纳入主仓库追踪 - 删除各自内部 .git 目录，消除嵌套 git 仓库
30 lines
787 B
Bash
30 lines
787 B
Bash
#!/bin/bash
|
||
#SBATCH --job-name=lejepa_scale
|
||
#SBATCH --output=logs/scale_%A_%a.out
|
||
#SBATCH --error=logs/scale_%A_%a.err
|
||
#SBATCH --partition=gpuq
|
||
#SBATCH --qos=slow_nice
|
||
#SBATCH --gres=gpu:v100:1
|
||
#SBATCH --cpus-per-task=4
|
||
#SBATCH --mem=16G
|
||
#SBATCH --time=12:00:00 # ← bumped from 6h: 3 modes × 5 seeds = 15 runs per N
|
||
#SBATCH --array=0-9 # 10 dims
|
||
|
||
DIMS=(2 4 8 16 32 64 128 256 512 1024)
|
||
SEEDS=(0 1 2 3 4)
|
||
MODES=(lejepa whiten infonce)
|
||
|
||
N=${DIMS[$SLURM_ARRAY_TASK_ID]}
|
||
|
||
eval "$(conda shell.bash hook)"
|
||
conda activate pytorch
|
||
|
||
mkdir -p logs
|
||
|
||
for MODE in "${MODES[@]}"; do
|
||
for SEED in "${SEEDS[@]}"; do
|
||
echo "N=${N} seed=${SEED} mode=${MODE}"
|
||
python -u run.py --config configs/scaling.yaml \
|
||
--N "${N}" --seed "${SEED}" --mode "${MODE}"
|
||
done
|
||
done |