----------------- Options --------------- ------------------- alpha: 0.25 [default: 0.25] batch_size_train: 1 [default: 1] batch_size_val: 1 [default: 1] beta: 0.9 [default: 0.9] curves: ['sweep_h', 'sweep_v', 'gilbert', 'gilbert_trans'] [default: ['sweep_h', 'sweep_v', 'gilbert', 'gilbert_trans']] de_att_window_size: None [default: None] de_attn_drop_rate: 0.0 [default: 0.0] de_bi_ssm: True [default: True] de_chunk_size: 256 [default: 256] de_conv_bias: False [default: False] de_curve_order: spatial [default: spatial] de_d_conv: [4] [default: [4]] de_d_state: [16] [default: [16]] de_depths: [1] [default: [1]] de_drop_path_rate: 0.0 [default: 0.0] de_drop_rate: 0.2 [default: 0.2] de_dt_init: random [default: random] de_dt_init_floor: 0.0001 [default: 0.0001] de_dt_max: 0.1 [default: 0.1] de_dt_min: 0.001 [default: 0.001] de_dt_rank: auto [default: auto] de_dt_scale: 1.0 [default: 1.0] de_embed_dim: [192] [default: [192]] de_embed_hres: 64 [default: 64] de_embed_norm: True [default: True] de_expand: [1] [default: [1]] de_grouping_size: [(1, 254945)] [default: [(1, 254945)]] de_headdim: 64 [default: 64] de_is_alibi_slopes: False [default: False] de_is_causal: False [default: False] de_is_deterministic: False [default: False] de_is_divide_out: True [default: True] de_is_out_norm: True [default: True] de_mlp_ratio: 1.0 [default: 1.0] de_n_heads: [4] [default: [4]] de_ngroups: 1 [default: 1] de_proj_bias: False [default: False] de_proj_drop: 0.0 [default: 0.0] de_qk_scale: None [default: None] de_qkv_bias: False [default: False] de_use_checkpoint: False [default: False] de_use_reentrant: False [default: False] decoder: Mamba [default: Mamba] delta_t: 4 [default: 4] delta_t_f: 7 [default: 7] en_att_window_size: None [default: None] en_attn_drop_rate: 0.0 [default: 0.0] en_bi_ssm: True [default: True] en_chunk_size: 256 [default: 256] en_conv_bias: False [default: False] en_curve_order: spatial_first [default: spatial_first] en_d_conv: [4, 4, 4] [default: [4, 4, 4]] en_d_state: [16, 16, 16] [default: [16, 16, 16]] en_depths: [2, 2, 2] [default: [2, 2, 2]] en_drop_path_rate: 0.0 [default: 0.0] en_drop_rate: 0.2 [default: 0.2] en_dt_init: random [default: random] en_dt_init_floor: 0.0001 [default: 0.0001] en_dt_max: 0.1 [default: 0.1] en_dt_min: 0.001 [default: 0.001] en_dt_rank: auto [default: auto] en_dt_scale: 1.0 [default: 1.0] en_embed_cpc: 16 [default: 16] en_embed_dim: [192, 192, 192] [default: [192, 192, 192]] en_embed_era5: 128 [default: 128] en_embed_glofas: 48 [default: 48] en_embed_norm: True [default: True] en_expand: [1, 1, 1] [default: [1, 1, 1]] en_grouping_size: [(4, 254945), (2, 254945), (1, 254945)] [default: [(4, 254945), (2, 254945), (1, 254945)]] en_headdim: 64 [default: 64] en_is_alibi_slopes: False [default: False] en_is_causal: False [default: False] en_is_deterministic: False [default: False] en_is_divide_out: True [default: True] en_is_out_norm: True [default: True] en_mlp_ratio: 1.0 [default: 1.0] en_n_heads: [3, 3, 3] [default: [3, 3, 3]] en_ngroups: 1 [default: 1] en_proj_bias: False [default: False] en_proj_drop: 0.0 [default: 0.0] en_qk_scale: None [default: None] en_qkv_bias: False [default: False] en_use_checkpoint: False [default: False] en_use_reentrant: False [default: False] encoder: Mamba [default: Mamba] gpu_id: 0, 1, 2, 3 [default: 0, 1, 2, 3] head: MLP [default: MLP] head_drop_rate: 0.1 [default: 0.1] head_hidden_dim: 64 [default: 64] head_out_dim: 1 [default: 1] head_use_checkpoint: False [default: False] head_use_reentrant: False [default: False] in_era5: 32 [default: 32] in_glofas: 4 [default: 4] in_hres: 7 [default: 7] in_static: 99 [default: 99] is_add_xyz: True [default: True] is_hres_forecast: True [default: True] is_mask_station: False [default: False] is_norm: True [default: True] is_obs: False [default: False] is_sample: True [default: True] is_sample_aifas: True [default: True] is_sample_curves: True [default: True] is_shuffle: False [default: False] is_shuffle_curves: False [default: False] lat_max: None [default: None] lat_min: None [default: None] lon_max: None [default: None] lon_min: None [default: None] lr: 0.0001 [default: 0.0001] lr_decay_rate: 0.9 [default: 0.9] lr_decay_step: 20 [default: 20] lr_min: 9e-05 [default: 9e-05] lr_scheduler: cosine [default: cosine] lr_warmup: 1e-06 [default: 1e-06] lr_warmup_epochs: 4 [default: 4] n_epochs: 20 [default: 20] n_points: 254945 [default: 254945] n_workers_train: 16 [default: 16] n_workers_val: 16 [default: 16] name: RiverMamba_aifas_reanalysis [default: RiverMamba_aifas_reanalysis] nan_fill: 0.0 [default: 0.0] optimizer: AdamW [default: AdamW] phase: train [default: train] pin_memory: True [default: True] pretrained_model: None [default: None] seed: 20 [default: 0] split_id: 0 [default: 0] static_dataset: LISFLOOD [default: LISFLOOD] training_checkpoint: None [default: None] val_every_n_epochs: 1 [default: 1] variables_hres_forecast_log1p: None [default: None] weight_decay: 0.0001 [default: 0.0001] years_train: ['2010', '2011', '2012', '2013', '2014', '2015', '2016', '2017', '2018'] [default: ['2010', '2011', '2012', '2013', '2014', '2015', '2016', '2017', '2018']] root_glofas_reanalysis: ./GloFAS_Reanalysis_Global [default: ./GloFAS_Reanalysis_Global] root_era5_land_reanalysis: ./ERA5-Land_Reanalysis_Global [default: ./ERA5-Land_Reanalysis_Global] root_hres_forecast: ./ECMWF_HRES_Global [default: ./ECMWF_HRES_Global] root_cpc: ./CPC_Global [default: ./CPC_Global] root_static: ./GloFAS_Static [default: ./GloFAS_Static] root_obs: ./GRDC_Obs_Global [default: ./GRDC_Obs_Global] dir_log: ./log [default: ./log] ----------------- Input Variables ------- ------------------- Variables GloFAS: ['acc_rod24', 'dis24', 'sd', 'swi'] Variables Era5-Land: ['d2m', 'e', 'es', 'evabs', 'evaow', 'evatc', 'evavt', 'lai_hv', 'lai_lv', 'pev', 'sf', 'skt', 'slhf', 'smlt', 'sp', 'src', 'sro', 'sshf', 'ssr', 'ssrd', 'ssro', 'stl1', 'str', 'strd', 'swvl1', 'swvl2', 'swvl3', 'swvl4', 't2m', 'tp', 'u10', 'v10'] Variables hres forecast: ['e', 'sf', 'sp', 'ssr', 'str', 't2m', 'tp'] Variables CPC: ['precip'] Variables Static: ['CalChanMan1', 'CalChanMan2', 'GwLoss', 'GwPercValue', 'LZTC', 'LZthreshold', 'LakeMultiplier', 'PowerPrefFlow', 'QSplitMult', 'ReservoirRnormqMult', 'SnowMeltCoef', 'UZTC', 'adjustNormalFlood', 'b_Xinanjiang', 'chanbnkf', 'chanbw', 'chanflpn', 'changrad', 'chanlength', 'chanman', 'cropcoef', 'cropgrpn', 'elv', 'elvstd', 'fracforest', 'fracgwused', 'fracirrigated', 'fracncused', 'fracother', 'fracrice', 'fracsealed', 'fracwater', 'genua1', 'genua2', 'genua3', 'gradient', 'gwbodies', 'ksat1', 'ksat2', 'ksat3', 'laif_01', 'laif_02', 'laif_03', 'laif_04', 'laif_05', 'laif_06', 'laif_07', 'laif_08', 'laif_09', 'laif_10', 'laif_11', 'laif_12', 'laii_01', 'laii_02', 'laii_03', 'laii_04', 'laii_05', 'laii_06', 'laii_07', 'laii_08', 'laii_09', 'laii_10', 'laii_11', 'laii_12', 'laio_01', 'laio_02', 'laio_03', 'laio_04', 'laio_05', 'laio_06', 'laio_07', 'laio_08', 'laio_09', 'laio_10', 'laio_11', 'laio_12', 'lambda1', 'lambda2', 'lambda3', 'ldd', 'mannings', 'pixarea', 'pixleng', 'riceharvestday1', 'riceharvestday2', 'riceharvestday3', 'riceplantingday1', 'riceplantingday2', 'riceplantingday3', 'soildepth2', 'soildepth3', 'thetas1', 'thetas2', 'thetas3', 'upArea', 'waterregions'] Variables GloFAS log1p: None Variables Era5-Land logp1: None Variables hres forecast logp1: None Variables CPC logp1: ['precip'] Variables Static logp1: ['chanbw', 'chanflpn', 'elvstd', 'ksat1', 'ksat2', 'ksat3', 'soildepth2', 'soildepth3', 'upArea', 'waterregions'] ----------------- Years ----------------- ------------------- Training: ['2010', '2011', '2012', '2013', '2014', '2015', '2016', '2017', '2018'] Validation: ['2019', '2020'] ----------------- End ------------------- -------------------