Develop the code to run for all the CAMELS basins

[RY4GIT]

• Currently, dCFE only supports NN for one CAMELS basin
• Goal: make it adaptable to parallel run with tensor expressions

[RY4GIT]

• [x] Dataloader and ParmaeterDict
• [x] Dataset class getitem() and len()
• [ ] CFE function
  • [x] BMI_CFE.initialize()
  • [x] ET from rainfall
  • [x] Soil classic
    • [x] ET
    • [x] Percolation
    • [x] Lateral flow
  • [ ] Partitioning
    • [x] Schaake
    • [ ] Xinanjiang
  • [ ] Soil ODE
  • [x] Groundwater
  • [x] GIUH
  • [x] Nash
  • [x] BMI_CFE.finalize()
• [ ] Debug
• [x] Synthetic output
• [x] Check gradient tracking

[RY4GIT]

Surface partitioning has mass balance issue

[RY4GIT]

Surface partitioning has mass balance issue

Fixed in e7c0af070b9424ca987200b5de37f76846bca314

[RY4GIT]

• Lateral and percolation flux looks too slow
• The dimension of reservoir['storage_m'] within the SM ode scheme changes at different timestep

[RY4GIT]

dimension issue solved

[RY4GIT]

Running without errors in 889840731d15fc60c5bc8cd099763c7cd929cedc