ckpt rho@400 Re_mu res@150 res@400 val /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 s600 0.97709 -0.2291 2.75e-01 9.57e-02 4.9611 s800 0.98169 -0.1831 5.13e-01 1.54e-01 4.2029 s1000 0.98278 -0.1722 3.45e+00 1.61e+00 3.9009 s1200 0.97454 -0.2546 6.77e-01 5.27e-02 3.6971 s1400 1.18812 +1.8812 1.78e+01 1.66e+01 3.6155 s1600 0.97216 -0.2784 4.56e-01 3.00e-02 3.3450 s1800 0.97043 -0.2959 2.63e+00 2.72e+00 3.2221 s2000 0.97087 -0.2913 2.58e+00 4.64e-02 3.1644 s2200 0.97327 -0.2674 3.80e+00 1.21e+00 3.0150 s2400 0.96983 -0.3017 2.90e+00 7.92e-02 2.9451 s2600 0.97321 -0.2680 3.75e+00 1.57e+00 2.8785 s2800 0.97357 -0.2686 5.54e+00 9.81e-01 2.8715 s3000 0.97211 -0.2789 3.99e+00 8.35e-02 2.7709 s3200 0.99111 -0.0889 1.15e+01 8.34e+00 2.7815 s3400 0.99111 -0.0889 1.15e+01 8.34e+00 2.7815 s3600 0.99111 -0.0889 1.15e+01 8.34e+00 2.7815 EIG_TRAJ_DONE