chore: OpenVINO MetricGAN GPU spike (do not merge)

Arc B580 lost the kill test: batch-14 mask p50 103 ms vs Torch CPU
70 ms enhance / OpenVINO CPU 5 ms. Production stays on known-good
CPU workers (tag known-good-cpu-2026-08-28). Hop stays 200 ms.
This commit is contained in:
root
2026-08-28 07:18:54 +00:00
parent 736eccb932
commit 42443420be
6 changed files with 307 additions and 0 deletions
+38
View File
@@ -0,0 +1,38 @@
#!/usr/bin/env python3
"""Dump MetricGAN enhance_model I/O so the OpenVINO spike can convert it."""
from __future__ import annotations
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
import torch
from speechbrain.inference.enhancement import SpectralMaskEnhancement
SAVEDIR = "/root/.cache/speechbrain-enhancement"
def main() -> None:
m = SpectralMaskEnhancement.from_hparams(
source="speechbrain/metricgan-plus-voicebank",
savedir=SAVEDIR,
run_opts={"device": "cpu"},
)
net = m.mods.enhance_model
print("enhance_model", type(net).__name__)
print("device", m.device)
wav = torch.randn(1, 16000)
feats = m.compute_features(wav)
print("wav", tuple(wav.shape), "feats", tuple(feats.shape), feats.dtype)
lengths = torch.tensor([1.0])
with torch.no_grad():
mask = net(feats, lengths=lengths)
print("mask", tuple(mask.shape), mask.dtype)
print("modules:")
for name, mod in net.named_children():
print(f" {name}: {type(mod).__name__}")
if __name__ == "__main__":
main()