=== s2000 ===
/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
  r=0.02  cos=0.9068 0.8533  mean=0.8801
  r=0.05  cos=0.9127 0.8550  mean=0.8839
  r=0.1   cos=0.9129 0.8573  mean=0.8851
  r=0.2   cos=0.9127 0.8573  mean=0.8850
  r=0.4   cos=0.9131 0.8573  mean=0.8852
=== fast-adaptive@2.18 ===
  r=0.02  cos=0.9242 0.8212  mean=0.8727
  r=0.05  cos=0.9245 0.8213  mean=0.8729
  r=0.1   cos=0.9242 0.8213  mean=0.8728
  r=0.2   cos=0.9242 0.8213  mean=0.8728
  r=0.4   cos=0.9243 0.8212  mean=0.8727
R_SWEEP_DONE