From 293785af4c9a556f388f04bcf9599aaee9e9dfd9 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 22 Jul 2026 16:47:55 -0500 Subject: experiment: add one-sided residual stability margin --- experiments/conv_local_smoke.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'experiments/conv_local_smoke.py') diff --git a/experiments/conv_local_smoke.py b/experiments/conv_local_smoke.py index 802051b..dbae759 100644 --- a/experiments/conv_local_smoke.py +++ b/experiments/conv_local_smoke.py @@ -994,6 +994,15 @@ def kp_mixed_traffic_checks(): assert closed_form["residual_traffic_rms_ratio"] < 1e-14 assert closed_form["max_absolute_residual_soma_slope"] < 1e-14 + for slope, bias in zip(net.P_traffic, net.P_traffic_bias): + slope.zero_() + bias.zero_() + stable_fit = net.predictor_closed_form_fit( + forward["hiddens"], stability_margin=1e-3) + assert stable_fit["max_positive_residual_soma_slope"] < 1e-14 + assert stable_fit["min_residual_soma_slope"] < -9e-4 + assert stable_fit["max_applied_stability_margin"] >= 1e-3 + for slope, bias in zip(net.P_traffic, net.P_traffic_bias): slope.zero_() bias.zero_() -- cgit v1.2.3