diff --git a/exponax/nonlin_fun/_general_nonlinear.py b/exponax/nonlin_fun/_general_nonlinear.py index 8b09cb9..8104dc1 100644 --- a/exponax/nonlin_fun/_general_nonlinear.py +++ b/exponax/nonlin_fun/_general_nonlinear.py @@ -89,6 +89,8 @@ def __init__( # Minus required because it internally has another minus scale=-scale_list[1], single_channel=True, + # For legacy reasons, the single-channel convection term is conservative + conservative=True, ) self.gradient_norm_nonlinear_fun = GradientNormNonlinearFun( num_spatial_dims,