bench: preset Kaggle batch sweep

HachimiMT_Benchmark_Profile.ipynb

@@ -23,7 +23,7 @@
     "BENCH_DONE ...\n",
     "```\n",
     "\n",
-    "Để so Kaggle với Colab công bằng, bật `USE_SINGLE_GPU_FOR_FAIR_TEST=True`; notebook sẽ tự dùng window `16` cho lần fair test nếu bạn không override `WINDOW_MULTIPLIER`."
+    "Preset hiện tại dùng để sweep batch trên Kaggle T4 x2: chạy cả 2 GPU, beam `[1, 2]`, batch `[64, 96, 128]`, window `[4]`. Để so Kaggle với Colab công bằng, bật `USE_SINGLE_GPU_FOR_FAIR_TEST=True`; notebook sẽ tự dùng window `16` cho lần fair test nếu bạn không override `WINDOW_MULTIPLIER`."
    ]
   },
   {
@@ -142,6 +142,10 @@
     "NORMALIZE = \"auto\"          # auto, t2s, none\n",
     "PROGRESS_SECONDS = 30\n",
     "\n",
+    "# Preset hiện tại ưu tiên batch sweep nên bỏ qua benchmark đơn khi Run all.\n",
+    "# Đổi thành True nếu bạn muốn chạy một cấu hình đơn trước sweep.\n",
+    "RUN_SINGLE_BENCHMARK = False\n",
+    "\n",
     "# Bật để so Kaggle x1 T4 công bằng với Colab x1 T4.\n",
     "# Tắt để Kaggle tự dùng toàn bộ GPU được cấp, ví dụ T4 x2.\n",
     "USE_SINGLE_GPU_FOR_FAIR_TEST = False\n",
@@ -326,7 +330,10 @@
     "    return summary\n",
     "\n",
     "\n",
-    "last_summary = run_benchmark_once()"
+    "if RUN_SINGLE_BENCHMARK:\n",
+    "    last_summary = run_benchmark_once()\n",
+    "else:\n",
+    "    print(\"RUN_SINGLE_BENCHMARK=False. Bỏ qua benchmark đơn, chạy tiếp cell sweep.\")"
    ]
   },
   {
@@ -335,13 +342,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# 5. Sweep batch/window/beam bằng script CLI (tắt mặc định)\n",
-    "# Bật RUN_SWEEP=True khi muốn tìm cấu hình tốt nhất trên runtime hiện tại.\n",
-    "RUN_SWEEP = False\n",
-    "SWEEP_SINGLE_GPU = True      # True: fair x1 T4; False: Kaggle dùng auto/all GPU\n",
+    "# 5. Sweep batch/window/beam bằng script CLI\n",
+    "# Preset hiện tại: Kaggle T4 x2 batch sweep.\n",
+    "RUN_SWEEP = True\n",
+    "SWEEP_SINGLE_GPU = False     # False: Kaggle dùng auto/all GPU; True: fair x1 T4\n",
     "SWEEP_BEAMS = [1, 2]\n",
-    "SWEEP_BATCHES = [96]         # mini-sweep; mở rộng thành [64, 96, 128] nếu cần\n",
-    "SWEEP_WINDOWS = [4, 8, 16]\n",
+    "SWEEP_BATCHES = [64, 96, 128]\n",
+    "SWEEP_WINDOWS = [4]          # Kaggle x2 hiện là window=4x/16x hiệu dụng\n",
     "SWEEP_PROGRESS_SECONDS = 999999\n",
     "SWEEP_MAX_RUNS = 0           # 0 = chạy hết combo; đặt 1/2 để smoke nhanh\n",
     "\n",
@@ -371,6 +378,8 @@
     "    ]\n",
     "    if SWEEP_SINGLE_GPU:\n",
     "        cmd.extend([\"--single-gpu\", \"--gpu-indices\", FAIR_GPU_INDICES])\n",
+    "    else:\n",
+    "        cmd.append(\"--auto-all-gpus\")\n",
     "    if TOKENIZE_WORKERS:\n",
     "        cmd.extend([\"--tokenize-workers\", TOKENIZE_WORKERS])\n",
     "    if TOKENIZE_JOB_SIZE:\n",

hachimimt-local.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d327ed12f4b4b829d254e5fef27dab272c9d6c61ad835c4f5f18f2ba97d35a94
-size 109385
+oid sha256:941b95caacc9c56acc8d24312736bad69d486ada6b09df95e6431c51aef56640
+size 109528