summaryrefslogtreecommitdiff
path: root/experiments
diff options
context:
space:
mode:
Diffstat (limited to 'experiments')
-rw-r--r--experiments/run.py2
-rw-r--r--experiments/smoke.py4
2 files changed, 6 insertions, 0 deletions
diff --git a/experiments/run.py b/experiments/run.py
index eb9f356..3cb2e00 100644
--- a/experiments/run.py
+++ b/experiments/run.py
@@ -115,6 +115,7 @@ def train(args):
if args.mode != "bp":
al = probes.alignment_report(net, px, py, poh, cfg)
rec["cos_r_negg"] = al["cos_r_negg"]
+ rec["cos_innovation_negg"] = al["cos_innovation_negg"]
rec["cos_apical_negg"] = al["cos_apical_negg"]
rec["cos_Ac_negg"] = al["cos_Ac_negg"]
rec["r_norm"] = al["r_norm"]
@@ -141,6 +142,7 @@ def train(args):
if args.mode != "bp":
al = probes.alignment_report(net, px, py, poh, cfg)
log["final"]["cos_r_negg"] = al["cos_r_negg"]
+ log["final"]["cos_innovation_negg"] = al["cos_innovation_negg"]
log["final"]["cos_apical_negg"] = al["cos_apical_negg"]
log["final"]["cos_Ac_negg"] = al["cos_Ac_negg"]
os.makedirs(args.outdir, exist_ok=True)
diff --git a/experiments/smoke.py b/experiments/smoke.py
index f440f7f..575f0bd 100644
--- a/experiments/smoke.py
+++ b/experiments/smoke.py
@@ -159,6 +159,10 @@ def main():
print(f" raw apical cos(a,-g) = {apical_cos:+.3f}")
print(f" pure A c cos(Ac,-g) = {sum(al_n['cos_Ac_negg'])/len(al_n['cos_Ac_negg']):+.3f}")
assert residual_cos > apical_cos + 0.1
+ cfg_raw = SDILConfig(use_residual=False, learn_A=False, learn_P=True)
+ al_raw = probes.alignment_report(net_n, x, y, yoh, cfg_raw)
+ assert al_raw["cos_r_negg"] == al_raw["cos_apical_negg"], (
+ "reported teaching alignment must honor use_residual=False")
# ---- CHECK 5: single-step loss-decrease ratio vs exact GD ----
print("\nCHECK5 single-step loss-decrease ratio vs exact GD:")