{
  "RUN_TAG": "multiH_JxxJyyJzz_fixedfields_S2_triplet_MixFunn_TDFNO",
  "device": "cuda",
  "manualSeed": 512,
  "n_qubits": 2,
  "Dim": 4,
  "Dim_eff": 3,
  "PARAM_NAMES": [
    "Jxx",
    "Jyy",
    "Jzz"
  ],
  "N_PARAMS_HAMILTONIAN": 3,
  "hx_fixed": 0.5,
  "hy_fixed": 0.5,
  "hz_fixed": 0.5,
  "PARAM_LOW": -1.0,
  "PARAM_HIGH": 1.0,
  "PARAM_SAMPLING": "sobol",
  "N_train_hamiltonians": 512,
  "GRID_POINTS_PER_AXIS": 9,
  "MAX_GRID_HAMILTONIANS": 512,
  "TEST_PARAMS": {
    "Jxx": 0.13,
    "Jyy": -0.37,
    "Jzz": 0.62
  },
  "T_train": 200,
  "t_train_max": 4.0,
  "dt_train": 0.02010050229728222,
  "AUTO_T_TEST_FROM_TRAIN_DT": true,
  "T_test": 598,
  "t_test_max": 12.0,
  "INITIAL_STATE_KIND": "zero",
  "CUSTOM_TRIPLET_COEFFS": [
    {
      "real": 1.0,
      "imag": 0.0
    },
    {
      "real": 1.0,
      "imag": 0.0
    },
    {
      "real": 1.0,
      "imag": 0.0
    }
  ],
  "opt": "LBFGS",
  "l_r": 1.0,
  "epochs": 550,
  "LBFGS_MAX_ITER": 30,
  "LBFGS_MAX_EVAL": 40,
  "LBFGS_TOL_GRAD": 0.0,
  "LBFGS_TOL_CHANGE": 0.0,
  "LBFGS_HISTORY_SIZE": 100,
  "LBFGS_LINE_SEARCH": null,
  "sched_type": "StepLR",
  "Restart": false,
  "use_data": true,
  "use_state_fidelity_loss": false,
  "LOSS_RESIDUAL_WEIGHT": 1.0,
  "LOSS_IC_WEIGHT": 1.0,
  "LOSS_NORM_WEIGHT": 1.0,
  "LOSS_DATA_WEIGHT": 1.0,
  "LOSS_STATE_FIDELITY_WEIGHT": 1.0,
  "INCLUDE_CORRELATIONS": true,
  "FULL_TWO_BODY": false,
  "obs_labels": [
    "X0",
    "Y0",
    "Z0",
    "X1",
    "Y1",
    "Z1",
    "X0X1",
    "Y0Y1",
    "Z0Z1"
  ],
  "max_s2_off_block_norm_train": 1.7156253306893632e-07,
  "s2_off_block_norm_test": 1.0065326350172654e-08,
  "USE_TIME_DOMAIN_FNO": true,
  "FNO_INPUT_SOURCE": "mixfunn",
  "FNO_TARGET_SOURCE": "exact",
  "FNO_USE_PARAMETER_CHANNELS": true,
  "FNO_USE_POSITIONAL_CHANNELS": true,
  "FNO_WINDOW_SIZE": 64,
  "FNO_SHIFT_SIZE": 32,
  "FNO_WINDOW_STRIDE": 8,
  "FNO_WIDTH": 32,
  "FNO_MODES": 16,
  "FNO_LAYERS": 3,
  "FNO_EPOCHS": 500,
  "FNO_BATCH_SIZE": 32,
  "FNO_LR": 0.001,
  "FNO_WEIGHT_DECAY": 1e-06,
  "FNO_LOSS_MSE_WEIGHT": 1.0,
  "FNO_LOSS_NORM_WEIGHT": 1.0,
  "FNO_LOSS_FIDELITY_WEIGHT": 1.0,
  "FNO_LOSS_SCHRODINGER_WEIGHT": 0.05,
  "FNO_LOSS_OVERLAP_WEIGHT": 5.0,
  "FNO_LOSS_SMOOTHNESS_WEIGHT": 0.01,
  "FNO_ROLLOUT_BLEND_OVERLAP": true,
  "FNO_COMPARE_BASIS_STATE_INDEX": 0
}