ep20 val_mae=[0.339, 0.445]
ep40 val_mae=[0.209, 0.279]
[rec_rrog_full_sig0.0_K1_bestq_T1_ns3_s0] test_mae=[0.213, 0.3] oracle@K=[0.213, 0.3] @ep40 (21.2s)
  wrote /home/yurenh2/rrog/runs/ckpt_rec_rrog_full_sig0.0_K1_bestq_T1_ns3_s0.pt