diff options
Diffstat (limited to 'sdil')
| -rw-r--r-- | sdil/conv.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/sdil/conv.py b/sdil/conv.py index d6bae4f..285ceb1 100644 --- a/sdil/conv.py +++ b/sdil/conv.py @@ -1267,8 +1267,8 @@ def conv_kp_mixed_traffic_step(net, x, y, config, step, rule, """One reciprocal-KP update using a selected mixed-apical signal.""" if not isinstance(net, CIFARKPMixedTrafficResNet): raise TypeError("mixed-traffic KP step requires CIFARKPMixedTrafficResNet") - if predictor_every < 1: - raise ValueError("predictor cadence must be positive") + if predictor_every < 0: + raise ValueError("predictor cadence must be nonnegative") config.validate() with torch.no_grad(): forward = net.forward( @@ -1303,7 +1303,8 @@ def conv_kp_mixed_traffic_step(net, x, y, config, step, rule, momentum=config.momentum, weight_decay=config.weight_decay, gamma_directions=gamma_directions, beta_directions=beta_directions) - did_predictor_update = step % predictor_every == 0 + did_predictor_update = ( + predictor_every > 0 and step % predictor_every == 0) predictor_mse = None if did_predictor_update: predictor_mse = net.predictor_step( |