diff options
Diffstat (limited to 'experiments/conv_local_smoke.py')
| -rw-r--r-- | experiments/conv_local_smoke.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/experiments/conv_local_smoke.py b/experiments/conv_local_smoke.py index 3267417..802051b 100644 --- a/experiments/conv_local_smoke.py +++ b/experiments/conv_local_smoke.py @@ -1075,6 +1075,17 @@ def kp_mixed_traffic_checks(): weight_decay=0.0, learn_A=False, learn_P=True), step=0, rule="innovation", predictor_every=16) assert math.isfinite(result["loss"]) and result["did_predictor_update"] + frozen_predictor = [value.clone() for value in + net.P_traffic + net.P_traffic_bias] + frozen_result = conv_kp_mixed_traffic_step( + net, x, y, ConvSDILConfig( + eta=1e-4, eta_output=1e-4, eta_P=0.1, momentum=0.0, + weight_decay=0.0, learn_A=False, learn_P=True), + step=1, rule="innovation", predictor_every=0) + assert math.isfinite(frozen_result["loss"]) + assert not frozen_result["did_predictor_update"] + assert all(torch.equal(before, after) for before, after in zip( + frozen_predictor, net.P_traffic + net.P_traffic_bias)) assert all(not value.requires_grad for value in net.W + net.Q + net.P_traffic + net.P_traffic_bias + [net.W_out, net.R_out, net.b_out]) |