mirror of
https://github.com/TaterTotterson/microWakeWord-Trainer-Nvidia-Docker.git
synced 2026-06-12 20:10:19 -06:00
Suppress Validation Batch in training
This commit is contained in:
@@ -854,6 +854,7 @@
|
||||
"source": [
|
||||
"# Train + export with GPU first, then automatic CPU fallback on GPU/VRAM errors\n",
|
||||
"# (LIVE streaming output + full log capture for error detection)\n",
|
||||
"# NOTE: Suppress ONLY the noisy \"Validation Batch #...\" lines (everything else still streams)\n",
|
||||
"import os, sys, subprocess, textwrap\n",
|
||||
"\n",
|
||||
"# ---- Common TF env (applies to BOTH attempts) ----\n",
|
||||
@@ -931,8 +932,14 @@
|
||||
" # Stream lines live AND capture them for OOM detection / error messages\n",
|
||||
" assert proc.stdout is not None\n",
|
||||
" for line in proc.stdout:\n",
|
||||
" print(line, end=\"\")\n",
|
||||
" full_log.append(line)\n",
|
||||
"\n",
|
||||
" # Hide ONLY the per-minibatch validation spam\n",
|
||||
" if line.startswith(\"Validation Batch #\"):\n",
|
||||
" continue\n",
|
||||
"\n",
|
||||
" # Everything else streams live\n",
|
||||
" print(line, end=\"\")\n",
|
||||
" finally:\n",
|
||||
" returncode = proc.wait()\n",
|
||||
"\n",
|
||||
|
||||
Reference in New Issue
Block a user