/home/yurenh2/miniconda3/lib/python3.13/site-packages/torch/autograd/graph.py:865: UserWarning: Attempting to run cuBLAS, but there was no current CUDA context! Attempting to set the primary context... (Triggered internally at /pytorch/aten/src/ATen/cuda/CublasHandlePool.cpp:330.)
  return Variable._execution_engine.run_backward(  # Calls into the C++ engine to run the backward pass
s1000  |l|=1.03113(mu=+0.1944+1.548j)  |l|=1.03113(mu=+0.1944-1.548j)  |l|=1.00482(mu=+0.0482+0.000j)
s1400  |l|=1.20728(mu=+2.0728+0.000j)  |l|=1.20201(mu=+2.0201+0.000j)  |l|=1.18788(mu=+1.8788+0.000j)
s2000  |l|=0.99798(mu=-0.0202+0.000j)  |l|=0.99788(mu=-0.0212+0.000j)  |l|=0.99753(mu=-0.0247+0.000j)
s2400  |l|=1.00559(mu=+0.0559+0.000j)  |l|=1.00018(mu=+0.0018+0.000j)  |l|=0.99530(mu=-0.0470+0.000j)
s2800  |l|=1.01613(mu=+0.1613+0.000j)  |l|=1.00169(mu=+0.0169+0.000j)  |l|=0.98866(mu=-0.4329+2.493j)
s3200  |l|=1.03371(mu=+0.1271-2.073j)  |l|=1.03371(mu=+0.1271+2.073j)  |l|=1.02563(mu=+0.1254-1.633j)
EIG_TRAJ2_DONE