Traceback (most recent call last):
  File "/projects/bfqt/users/yurenh2/ml-projects/snn-training/files/experiments/lyapunov_diffonly_benchmark.py", line 590, in <module>
    run_benchmark()
  File "/projects/bfqt/users/yurenh2/ml-projects/snn-training/files/experiments/lyapunov_diffonly_benchmark.py", line 497, in run_benchmark
    logits_diff, lyap_diff = model_diff(x, compute_lyapunov=True, lyap_eps=1e-4)
  File "/u/yurenh2/.local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1773, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/u/yurenh2/.local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1784, in _call_impl
    return forward_call(*args, **kwargs)
  File "/projects/bfqt/users/yurenh2/ml-projects/snn-training/files/experiments/lyapunov_diffonly_benchmark.py", line 307, in forward
    h_perturbed = h_base + h_delta
RuntimeError: The size of tensor a (16) must match the size of tensor b (8) at non-singleton dimension 3