diff options
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) |