#!/bin/bash

# === 修改这里 ===
# 1. 你的 Sweep ID (从 wandb sweep sweep.yaml 命令的输出中获得)
SWEEP_ID="1217820711-sun-yat-sen-university/cifar10_chanllenge/srzfvp0g"

# 2. 你想开几个 Agent (并行数)
# 你的 9800X3D + 5070Ti 建议开 3 个
NUM_AGENTS=2

# 3. Tmux 会话名称 (随便起)
SESSION_NAME="sweep_resnet18_try1"

# 4. 你的 Conda 环境名
CONDA_ENV="deep_learning"
# ===============

# 检查是否已经存在同名会话，如果有，先杀掉 (防止报错)
tmux has-session -t $SESSION_NAME 2>/dev/null
if [ $? == 0 ]; then
    echo "⚠️  Session $SESSION_NAME already exists. Killing it..."
    tmux kill-session -t $SESSION_NAME
fi

# 创建新会话 (后台模式)
tmux new-session -d -s $SESSION_NAME

# 循环创建窗口并运行 Agent
for ((i=1; i<=NUM_AGENTS; i++)); do
    # 如果不是第一个，就切分屏幕
    if [ $i -gt 1 ]; then
        tmux split-window -t $SESSION_NAME
        tmux select-layout -t $SESSION_NAME tiled
    fi

    # 发送命令：激活环境 -> 运行 Agent
    # C-m 代表回车键
    tmux send-keys -t $SESSION_NAME "conda activate $CONDA_ENV" C-m
    tmux send-keys -t $SESSION_NAME "wandb agent $SWEEP_ID" C-m
    
    echo "🚀 Agent $i started..."
done

# 进入 Tmux 界面
echo "All agents running! Attaching..."
tmux attach -t $SESSION_NAME