diff options
Diffstat (limited to 'experiments/run.py')
| -rw-r--r-- | experiments/run.py | 2 |
1 files changed, 2 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) |