diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-21 09:56:50 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-21 09:56:50 -0500 |
| commit | dd400bbe1987b5c35ca8e3a1034cb48bc3c93aec (patch) | |
| tree | 717b6d53fe424214d9a2bd1385abf0bbd70dc55b /experiments/synthetic_smoke.py | |
| parent | e07ee19312f462d204aa48e77e739abd102bcabb (diff) | |
feat: add sequential FA to scaling runner
Diffstat (limited to 'experiments/synthetic_smoke.py')
| -rw-r--r-- | experiments/synthetic_smoke.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/experiments/synthetic_smoke.py b/experiments/synthetic_smoke.py index c118813..cd6e2af 100644 --- a/experiments/synthetic_smoke.py +++ b/experiments/synthetic_smoke.py @@ -26,7 +26,7 @@ def check_task(dataset, expected_in, expected_out, extra): x, y = next(iter(train)) assert x.shape == (16, n_in) assert y.min() >= 0 and y.max() < n_out - for mode in ("bp", "dfa", "sdil"): + for mode in ("bp", "fa", "dfa", "sdil"): args.mode = mode net, _ = build(args, "cpu") assert net.logits(x).shape == (16, n_out) |