Add files using upload-large-folder tool

day6/baseline_rmse_bias_nogrid_18heads_res2next50_20251228_1227/config.json

@@ -0,0 +1,112 @@
+{
+  "name": "baseline_rmse_bias_nogrid_18heads",
+  "description": "R2N50 with RMSE biased loss and no grid with 18 stereo heads",
+  "tags": [
+    "baseline",
+    "res2next50"
+  ],
+  "paths": {
+    "data_dir": "/csiro-biomass_data/",
+    "train_csv": "train.csv",
+    "test_csv": "test.csv",
+    "train_images": "train",
+    "test_images": "test",
+    "output_dir": "/volume/exp/day6"
+  },
+  "model": {
+    "backbone": "timm/res2next50.in1k",
+    "pretrained": true,
+    "freeze_percent": 0.3,
+    "img_size": 224,
+    "grid": [
+      4,
+      4
+    ],
+    "use_stereo": true,
+    "stereo_heads": 16,
+    "dropout": 0.3,
+    "use_global": true,
+    "global_weight_learnable": true,
+    "head_strategy": "derive",
+    "use_dead_ratio_head": true,
+    "use_film": false,
+    "film_hidden_dim": null,
+    "film_hidden_ratio": 0.5,
+    "film_dropout": 0.0,
+    "film_share": false,
+    "film_apply_to_global": false
+  },
+  "data": {
+    "batch_size": 12,
+    "num_workers": 4,
+    "pin_memory": true,
+    "use_augmentation": true,
+    "aug_hflip": 0.3,
+    "aug_vflip": 0.3,
+    "aug_rotate90": 0.3,
+    "aug_brightness_contrast": 0.2,
+    "aug_hue_sat": 0.0,
+    "aug_noise": 0.1
+  },
+  "fold": {
+    "n_folds": 1,
+    "fold_idx": null,
+    "stratify": true,
+    "stratify_column": "State",
+    "stratify_bins": 5,
+    "shuffle": true,
+    "test_size": 0.2
+  },
+  "loss": {
+    "loss_type": "rmse",
+    "huber_delta": 1.0,
+    "weight_total": 0.3,
+    "weight_gdm": 0.2,
+    "weight_green": 0.2,
+    "weight_clover": 0.2,
+    "weight_dead": 0.2
+  },
+  "optimizer": {
+    "optimizer": "adamw",
+    "learning_rate": "4e-5",
+    "weight_decay": 0.01,
+    "scheduler": "onecycle",
+    "warmup_epochs": 1,
+    "warmup_pct": 0.1,
+    "min_lr": 1e-07,
+    "gradient_accumulation_steps": 1,
+    "max_grad_norm": 0.0
+  },
+  "training": {
+    "epochs": 26,
+    "seed": 42,
+    "mixed_precision": true,
+    "patience": 7,
+    "early_stopping_metric": "val_weighted_r2",
+    "early_stopping_mode": "max",
+    "save_best_only": true,
+    "save_last": true
+  },
+  "aux": {
+    "enabled": true,
+    "predict_height": true,
+    "height_target": "raw",
+    "height_weight": 0.05,
+    "predict_ndvi": true,
+    "ndvi_target": "raw",
+    "ndvi_weight": 0.05,
+    "predict_state": true,
+    "state_weight": 0.03,
+    "state_num_classes": 4,
+    "state_class_weighting": "balanced",
+    "state_class_weight_max": 10.0,
+    "state_effective_num_beta": 0.999,
+    "predict_species": false,
+    "species_weight": 0.03,
+    "species_num_classes": null,
+    "species_class_weighting": "balanced",
+    "species_class_weight_max": 10.0,
+    "species_effective_num_beta": 0.999,
+    "label_smoothing": 0.015
+  }
+}

day6/baseline_rmse_bias_nogrid_18heads_res2next50_20251228_1227/config.yaml

@@ -0,0 +1,100 @@
+name: baseline_rmse_bias_nogrid_18heads
+description: R2N50 with RMSE biased loss and no grid with 18 stereo heads
+tags:
+- baseline
+- res2next50
+paths:
+  data_dir: /csiro-biomass_data/
+  train_csv: train.csv
+  test_csv: test.csv
+  train_images: train
+  test_images: test
+  output_dir: /volume/exp/day6
+model:
+  backbone: timm/res2next50.in1k
+  pretrained: true
+  freeze_percent: 0.3
+  img_size: 224
+  grid:
+  - 4
+  - 4
+  use_stereo: true
+  stereo_heads: 16
+  dropout: 0.3
+  use_global: true
+  global_weight_learnable: true
+  head_strategy: derive
+  use_dead_ratio_head: true
+  use_film: false
+  film_hidden_dim: null
+  film_hidden_ratio: 0.5
+  film_dropout: 0.0
+  film_share: false
+  film_apply_to_global: false
+data:
+  batch_size: 12
+  num_workers: 4
+  pin_memory: true
+  use_augmentation: true
+  aug_hflip: 0.3
+  aug_vflip: 0.3
+  aug_rotate90: 0.3
+  aug_brightness_contrast: 0.2
+  aug_hue_sat: 0.0
+  aug_noise: 0.1
+fold:
+  n_folds: 1
+  fold_idx: null
+  stratify: true
+  stratify_column: State
+  stratify_bins: 5
+  shuffle: true
+  test_size: 0.2
+loss:
+  loss_type: rmse
+  huber_delta: 1.0
+  weight_total: 0.3
+  weight_gdm: 0.2
+  weight_green: 0.2
+  weight_clover: 0.2
+  weight_dead: 0.2
+optimizer:
+  optimizer: adamw
+  learning_rate: 4e-5
+  weight_decay: 0.01
+  scheduler: onecycle
+  warmup_epochs: 1
+  warmup_pct: 0.1
+  min_lr: 1.0e-07
+  gradient_accumulation_steps: 1
+  max_grad_norm: 0.0
+training:
+  epochs: 26
+  seed: 42
+  mixed_precision: true
+  patience: 7
+  early_stopping_metric: val_weighted_r2
+  early_stopping_mode: max
+  save_best_only: true
+  save_last: true
+aux:
+  enabled: true
+  predict_height: true
+  height_target: raw
+  height_weight: 0.05
+  predict_ndvi: true
+  ndvi_target: raw
+  ndvi_weight: 0.05
+  predict_state: true
+  state_weight: 0.03
+  state_num_classes: 4
+  state_class_weighting: balanced
+  state_class_weight_max: 10.0
+  state_effective_num_beta: 0.999
+  predict_species: false
+  species_weight: 0.03
+  species_num_classes: null
+  species_class_weighting: balanced
+  species_class_weight_max: 10.0
+  species_effective_num_beta: 0.999
+  label_smoothing: 0.015

day6/baseline_rmse_nobias_4grid_16heads_f30_aux_res2next50_20251228_1235/config.json

@@ -0,0 +1,112 @@
+{
+  "name": "baseline_rmse_nobias_4grid_16heads_f30_aux",
+  "description": "R2N50 with RMSE biased loss and 4 grid with 18 stereo heads",
+  "tags": [
+    "baseline",
+    "res2next50"
+  ],
+  "paths": {
+    "data_dir": "/csiro-biomass_data/",
+    "train_csv": "train.csv",
+    "test_csv": "test.csv",
+    "train_images": "train",
+    "test_images": "test",
+    "output_dir": "/volume/exp/day6"
+  },
+  "model": {
+    "backbone": "timm/res2next50.in1k",
+    "pretrained": true,
+    "freeze_percent": 0.3,
+    "img_size": 224,
+    "grid": [
+      4,
+      4
+    ],
+    "use_stereo": true,
+    "stereo_heads": 16,
+    "dropout": 0.3,
+    "use_global": true,
+    "global_weight_learnable": true,
+    "head_strategy": "derive",
+    "use_dead_ratio_head": true,
+    "use_film": false,
+    "film_hidden_dim": null,
+    "film_hidden_ratio": 0.5,
+    "film_dropout": 0.0,
+    "film_share": false,
+    "film_apply_to_global": false
+  },
+  "data": {
+    "batch_size": 12,
+    "num_workers": 4,
+    "pin_memory": true,
+    "use_augmentation": true,
+    "aug_hflip": 0.3,
+    "aug_vflip": 0.3,
+    "aug_rotate90": 0.3,
+    "aug_brightness_contrast": 0.2,
+    "aug_hue_sat": 0.0,
+    "aug_noise": 0.1
+  },
+  "fold": {
+    "n_folds": 1,
+    "fold_idx": null,
+    "stratify": true,
+    "stratify_column": "State",
+    "stratify_bins": 5,
+    "shuffle": true,
+    "test_size": 0.2
+  },
+  "loss": {
+    "loss_type": "rmse",
+    "huber_delta": 1.0,
+    "weight_total": 0.3,
+    "weight_gdm": 0.2,
+    "weight_green": 0.2,
+    "weight_clover": 0.2,
+    "weight_dead": 0.2
+  },
+  "optimizer": {
+    "optimizer": "adamw",
+    "learning_rate": "4e-5",
+    "weight_decay": 0.01,
+    "scheduler": "onecycle",
+    "warmup_epochs": 1,
+    "warmup_pct": 0.1,
+    "min_lr": 1e-07,
+    "gradient_accumulation_steps": 1,
+    "max_grad_norm": 0.0
+  },
+  "training": {
+    "epochs": 26,
+    "seed": 42,
+    "mixed_precision": true,
+    "patience": 7,
+    "early_stopping_metric": "val_weighted_r2",
+    "early_stopping_mode": "max",
+    "save_best_only": true,
+    "save_last": true
+  },
+  "aux": {
+    "enabled": true,
+    "predict_height": true,
+    "height_target": "raw",
+    "height_weight": 0.05,
+    "predict_ndvi": true,
+    "ndvi_target": "norm",
+    "ndvi_weight": 0.05,
+    "predict_state": true,
+    "state_weight": 0.03,
+    "state_num_classes": 4,
+    "state_class_weighting": "balanced",
+    "state_class_weight_max": 10.0,
+    "state_effective_num_beta": 0.999,
+    "predict_species": false,
+    "species_weight": 0.03,
+    "species_num_classes": null,
+    "species_class_weighting": "balanced",
+    "species_class_weight_max": 10.0,
+    "species_effective_num_beta": 0.999,
+    "label_smoothing": 0.015
+  }
+}

day6/baseline_rmse_nobias_4grid_16heads_f30_aux_res2next50_20251228_1235/config.yaml

@@ -0,0 +1,100 @@
+name: baseline_rmse_nobias_4grid_16heads_f30_aux
+description: R2N50 with RMSE biased loss and 4 grid with 18 stereo heads
+tags:
+- baseline
+- res2next50
+paths:
+  data_dir: /csiro-biomass_data/
+  train_csv: train.csv
+  test_csv: test.csv
+  train_images: train
+  test_images: test
+  output_dir: /volume/exp/day6
+model:
+  backbone: timm/res2next50.in1k
+  pretrained: true
+  freeze_percent: 0.3
+  img_size: 224
+  grid:
+  - 4
+  - 4
+  use_stereo: true
+  stereo_heads: 16
+  dropout: 0.3
+  use_global: true
+  global_weight_learnable: true
+  head_strategy: derive
+  use_dead_ratio_head: true
+  use_film: false
+  film_hidden_dim: null
+  film_hidden_ratio: 0.5
+  film_dropout: 0.0
+  film_share: false
+  film_apply_to_global: false
+data:
+  batch_size: 12
+  num_workers: 4
+  pin_memory: true
+  use_augmentation: true
+  aug_hflip: 0.3
+  aug_vflip: 0.3
+  aug_rotate90: 0.3
+  aug_brightness_contrast: 0.2
+  aug_hue_sat: 0.0
+  aug_noise: 0.1
+fold:
+  n_folds: 1
+  fold_idx: null
+  stratify: true
+  stratify_column: State
+  stratify_bins: 5
+  shuffle: true
+  test_size: 0.2
+loss:
+  loss_type: rmse
+  huber_delta: 1.0
+  weight_total: 0.3
+  weight_gdm: 0.2
+  weight_green: 0.2
+  weight_clover: 0.2
+  weight_dead: 0.2
+optimizer:
+  optimizer: adamw
+  learning_rate: 4e-5
+  weight_decay: 0.01
+  scheduler: onecycle
+  warmup_epochs: 1
+  warmup_pct: 0.1
+  min_lr: 1.0e-07
+  gradient_accumulation_steps: 1
+  max_grad_norm: 0.0
+training:
+  epochs: 26
+  seed: 42
+  mixed_precision: true
+  patience: 7
+  early_stopping_metric: val_weighted_r2
+  early_stopping_mode: max
+  save_best_only: true
+  save_last: true
+aux:
+  enabled: true
+  predict_height: true
+  height_target: raw
+  height_weight: 0.05
+  predict_ndvi: true
+  ndvi_target: norm
+  ndvi_weight: 0.05
+  predict_state: true
+  state_weight: 0.03
+  state_num_classes: 4
+  state_class_weighting: balanced
+  state_class_weight_max: 10.0
+  state_effective_num_beta: 0.999
+  predict_species: false
+  species_weight: 0.03
+  species_num_classes: null
+  species_class_weighting: balanced
+  species_class_weight_max: 10.0
+  species_effective_num_beta: 0.999
+  label_smoothing: 0.015

day6/baseline_rmse_nobias_4grid_16heads_f30_aux_res2next50_20251228_1235/final_results.json

@@ -0,0 +1,330 @@
+{
+  "experiment_name": "baseline_rmse_nobias_4grid_16heads_f30_aux",
+  "timestamp": "2025-12-28T12:44:04.304911",
+  "n_folds": 1,
+  "best_r2_mean": 0.6750052490640381,
+  "best_r2_std": 0.0,
+  "best_r2_max": 0.6750052490640381,
+  "best_r2_min": 0.6750052490640381,
+  "final_val_r2_mean": 0.5835581005272565,
+  "final_val_r2_std": 0.0,
+  "final_val_legacy_r2_mean": 0.4125139713287354,
+  "final_val_legacy_r2_std": 0.0,
+  "fold_results": [
+    {
+      "fold": 0,
+      "best_r2": 0.6750052490640381,
+      "best_epoch": 7,
+      "final_train_r2": 0.8211142386416808,
+      "final_val_r2": 0.5835581005272565,
+      "final_train_legacy_r2": 0.7223526775836945,
+      "final_val_legacy_r2": 0.4125139713287354,
+      "history": {
+        "train_loss": [
+          44.67302944349206,
+          41.23853442979895,
+          30.481522850368332,
+          22.22895273954972,
+          20.106135700059973,
+          18.555989389834195,
+          17.82048382966415,
+          16.19447836668595,
+          14.470834980840268,
+          14.156427445619002,
+          14.296914349431576,
+          12.601583667423414,
+          12.921115999636443,
+          12.208177297011666
+        ],
+        "val_loss": [
+          40.96238136291504,
+          34.23955758412679,
+          21.458637555440266,
+          22.398822784423828,
+          21.34282620747884,
+          17.182916323343914,
+          14.74948263168335,
+          15.683491865793863,
+          16.40652847290039,
+          17.321360905965168,
+          15.607057253519693,
+          14.620096524556478,
+          14.935823122660318,
+          16.447180430094402
+        ],
+        "train_r2": [
+          -1.297984987448133,
+          -1.110646311684088,
+          -0.06237562311266309,
+          0.5072216140193049,
+          0.5890377844311987,
+          0.6265540445704672,
+          0.6941320990975297,
+          0.6624401059018867,
+          0.7698876791198309,
+          0.7424350394218671,
+          0.7417317661333154,
+          0.8088259328902726,
+          0.8096646331704598,
+          0.8211142386416808
+        ],
+        "val_r2": [
+          -1.7197865475456213,
+          -0.9048272161926585,
+          0.38920085565438156,
+          0.3863712210604704,
+          0.35651746639992055,
+          0.5699174720814089,
+          0.6750052490640381,
+          0.6279995564747828,
+          0.5720505438273419,
+          0.5102895471428893,
+          0.600229547938933,
+          0.6434009290585221,
+          0.6528085922155441,
+          0.5835581005272565
+        ],
+        "train_legacy_r2": [
+          -1.710973572731018,
+          -1.4943776965141298,
+          -0.3457312703132629,
+          0.299228310585022,
+          0.4022741436958313,
+          0.4485288023948669,
+          0.533272123336792,
+          0.5149783134460449,
+          0.6288481473922729,
+          0.6204493582248688,
+          0.6234737932682037,
+          0.6989212810993195,
+          0.7016489923000336,
+          0.7223526775836945
+        ],
+        "val_legacy_r2": [
+          -2.5646701097488402,
+          -1.5498529911041259,
+          0.09964157342910768,
+          0.09868519902229309,
+          0.07493993043899536,
+          0.35032897591590884,
+          0.4968108892440796,
+          0.4361003220081329,
+          0.37986122369766234,
+          0.31046568751335146,
+          0.42186597585678104,
+          0.48759667873382573,
+          0.49459387063980104,
+          0.4125139713287354
+        ],
+        "train_mean_r2": [
+          -1.2196917057037353,
+          -1.0580377340316773,
+          -0.27132511138916016,
+          0.21799101829528808,
+          0.3211099743843079,
+          0.3680118203163147,
+          0.43383642435073855,
+          0.45482985973358153,
+          0.5311375260353088,
+          0.5613609433174134,
+          0.5713423967361451,
+          0.6296806812286377,
+          0.6361116766929626,
+          0.6608503222465515
+        ],
+        "val_mean_r2": [
+          -1.774488353729248,
+          -1.098954677581787,
+          0.07402974367141724,
+          0.0739555835723877,
+          0.09010010957717896,
+          0.315220844745636,
+          0.45743632316589355,
+          0.4198712944984436,
+          0.40418959856033326,
+          0.37601072788238527,
+          0.45030262470245364,
+          0.5200178027153015,
+          0.5064591407775879,
+          0.44728662967681887
+        ],
+        "lr": [
+          1.340511738248337e-05,
+          3.5013337543846056e-05,
+          3.997116856458639e-05,
+          3.964776883812336e-05,
+          3.897077000978936e-05,
+          3.7952356461798174e-05,
+          3.661085723609364e-05,
+          3.4970416154823484e-05,
+          3.306055728848372e-05,
+          3.0915653592281945e-05,
+          2.857430827401197e-05,
+          2.6078660027358885e-05,
+          2.347362463479648e-05,
+          2.0806086589436224e-05
+        ],
+        "train_Dry_Green_g": [
+          -0.9755585193634033,
+          -0.9225836992263794,
+          -0.16270995140075684,
+          0.4016740918159485,
+          0.5384881496429443,
+          0.580978512763977,
+          0.5929896235466003,
+          0.6021211743354797,
+          0.7116236686706543,
+          0.7040186524391174,
+          0.7083362340927124,
+          0.78301602602005,
+          0.8028672933578491,
+          0.7834655046463013
+        ],
+        "val_Dry_Green_g": [
+          -1.3394179344177246,
+          -0.9100496768951416,
+          0.15072131156921387,
+          0.10180026292800903,
+          0.1487189531326294,
+          0.42252177000045776,
+          0.596325159072876,
+          0.6022237539291382,
+          0.5454281568527222,
+          0.5375401973724365,
+          0.5914433002471924,
+          0.6618789434432983,
+          0.6085689067840576,
+          0.4828593134880066
+        ],
+        "train_Dry_Clover_g": [
+          -0.3051319122314453,
+          -0.1321810483932495,
+          -0.07221317291259766,
+          -0.05824780464172363,
+          0.003586113452911377,
+          0.11875057220458984,
+          0.18609493970870972,
+          0.29084646701812744,
+          0.25898420810699463,
+          0.4163244962692261,
+          0.4892765283584595,
+          0.5243823528289795,
+          0.5295500755310059,
+          0.6218262910842896
+        ],
+        "val_Dry_Clover_g": [
+          -0.2604970932006836,
+          -0.00041556358337402344,
+          -0.03519999980926514,
+          -0.032502174377441406,
+          0.075927734375,
+          0.19931352138519287,
+          0.3372766971588135,
+          0.32731759548187256,
+          0.46644169092178345,
+          0.4978267550468445,
+          0.5199596881866455,
+          0.6525763869285583,
+          0.5788671970367432,
+          0.63451087474823
+        ],
+        "train_Dry_Dead_g": [
+          -0.8752574920654297,
+          -0.8594757318496704,
+          -0.4895132780075073,
+          -0.010583043098449707,
+          0.07648348808288574,
+          0.03566664457321167,
+          0.13171273469924927,
+          0.21736866235733032,
+          0.20284318923950195,
+          0.29218506813049316,
+          0.2854683995246887,
+          0.27190953493118286,
+          0.3031235337257385,
+          0.3242262601852417
+        ],
+        "val_Dry_Dead_g": [
+          -1.3177006244659424,
+          -1.3177006244659424,
+          0.02978801727294922,
+          0.028744816780090332,
+          0.10440647602081299,
+          0.15156340599060059,
+          0.25202476978302,
+          0.19301944971084595,
+          0.21113348007202148,
+          0.24730443954467773,
+          0.2419680953025818,
+          0.3143344521522522,
+          0.33884239196777344,
+          0.32163870334625244
+        ],
+        "train_Dry_Total_g": [
+          -2.356255531311035,
+          -2.0925467014312744,
+          -0.4894993305206299,
+          0.38173872232437134,
+          0.47673314809799194,
+          0.5135218501091003,
+          0.635051429271698,
+          0.5706069469451904,
+          0.7168524265289307,
+          0.6678041219711304,
+          0.6681317090988159,
+          0.75723797082901,
+          0.7636380195617676,
+          0.7815138697624207
+        ],
+        "val_Dry_Total_g": [
+          -3.606477737426758,
+          -2.2457163333892822,
+          0.13380920886993408,
+          0.1151130199432373,
+          0.05915045738220215,
+          0.37482666969299316,
+          0.5264574289321899,
+          0.42828369140625,
+          0.3265731930732727,
+          0.20907366275787354,
+          0.3563348650932312,
+          0.4348595142364502,
+          0.46920859813690186,
+          0.363760769367218
+        ],
+        "train_GDM_g": [
+          -1.5862550735473633,
+          -1.2834014892578125,
+          -0.14268982410430908,
+          0.37537312507629395,
+          0.5102589726448059,
+          0.5911415219306946,
+          0.6233333945274353,
+          0.5932060480117798,
+          0.7653841376304626,
+          0.7264723777770996,
+          0.7054991126060486,
+          0.8118575215339661,
+          0.7813794612884521,
+          0.7932196855545044
+        ],
+        "val_GDM_g": [
+          -2.348348379135132,
+          -1.0208911895751953,
+          0.09103018045425415,
+          0.1566219925880432,
+          0.062296926975250244,
+          0.42787885665893555,
+          0.5750975608825684,
+          0.5485119819641113,
+          0.47137147188186646,
+          0.388308584690094,
+          0.5418071746826172,
+          0.5364397168159485,
+          0.5368086099624634,
+          0.4336634874343872
+        ]
+      }
+    }
+  ]
+}

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/binned_metrics.csv

@@ -0,0 +1,16 @@
+Target,Bin,N,R2,MAE,RMSE,Mean_True,Mean_Pred,Bias
+Dry_Green_g,Low (0-25%),29,-14.152999877929688,6.774202346801758,11.226171052674472,3.3235552,9.29852,5.974965
+Dry_Green_g,Medium (25-75%),60,-0.16533446311950684,5.732696533203125,7.306908711417721,20.724438,21.52433,0.7998924
+Dry_Green_g,High (75-100%),29,0.2658150792121887,12.945832252502441,16.324790329338843,52.78018,50.344288,-2.435894
+Dry_Clover_g,Low (0-25%),46,0.0,0.5260372161865234,1.3568069052950111,0.0,0.5260372,0.5260372
+Dry_Clover_g,Medium (25-75%),43,-2.417848587036133,2.869246244430542,3.645020414276936,2.723507,4.152144,1.428637
+Dry_Clover_g,High (75-100%),29,-0.5537570714950562,11.128981590270996,15.544881506928675,21.493078,10.91498,-10.578098
+Dry_Dead_g,Low (0-25%),29,-31.68956756591797,4.350475311279297,6.419685083193942,1.177231,5.194203,4.016972
+Dry_Dead_g,Medium (25-75%),60,-0.7690467834472656,4.875683307647705,5.99966334352151,8.776185,11.013793,2.237608
+Dry_Dead_g,High (75-100%),29,-2.100958824157715,14.734441757202148,19.43613323545824,29.732954,15.08216,-14.650794
+Dry_Total_g,Low (0-25%),29,-1.2302138805389404,6.716701507568359,8.425259676137678,16.129137,21.703938,5.5748005
+Dry_Total_g,Medium (25-75%),60,-0.3435556888580322,7.3824543952941895,9.956433571214534,39.200268,37.18724,-2.0130272
+Dry_Total_g,High (75-100%),29,-0.5880200862884521,20.144737243652344,25.228040758793455,79.825905,63.501076,-16.32483
+GDM_g,Low (0-25%),29,-0.523722767829895,3.6658477783203125,5.008262863975178,10.444342,11.966117,1.5217752
+GDM_g,Medium (25-75%),60,-1.1843562126159668,6.3629302978515625,9.776765910686585,26.978785,26.859766,-0.119018555
+GDM_g,High (75-100%),29,0.2878414988517761,12.093626022338867,15.328558026184634,60.158333,51.67322,-8.485111

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/metrics.csv

@@ -0,0 +1,6 @@
+Target,R2,MAE,RMSE,Slope,Intercept,Pred_Var,True_Var,Var_Ratio,Shrinkage
+Dry_Green_g,0.7184714078903198,8.088805198669434,11.871130383087603,0.7480752685565998,7.203156320598122,391.8102,504.8085,0.7761561,0.2519247314434002
+Dry_Clover_g,0.4424739480018616,4.258756637573242,8.693744418638126,0.34990625243998336,2.1822451570342785,30.455727,136.71416,0.22276936,0.6500937475600166
+Dry_Dead_g,0.19017571210861206,7.663343906402588,12.52137698465296,0.2088793715505795,8.000061103333557,40.33029,195.24431,0.20656322,0.7911206284494205
+Dry_Total_g,0.6339830160140991,10.964424133300781,16.73461116050682,0.5800940651759872,14.344788321991853,387.10733,771.6048,0.50169116,0.41990593482401284
+GDM_g,0.7276959419250488,7.477631568908691,11.552332020941499,0.7342980200772142,6.267061316641904,361.37445,494.254,0.7311513,0.26570197992278577

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/regression_stats.csv

@@ -0,0 +1,6 @@
+Target,Slope,Intercept,R_squared,Shrinkage,Mean_True,Mean_Pred
+Dry_Green_g,0.7480752685565998,7.203156320598122,0.7210103497236385,0.2519247314434002,25.126677,25.999802
+Dry_Clover_g,0.34990625243998336,2.1822451570342785,0.549601554983877,0.6500937475600166,6.686807,4.522001
+Dry_Dead_g,0.2088793715505795,8.000061103333557,0.2112214908682081,0.7911206284494205,12.662247,10.644943
+Dry_Total_g,0.5800940651759872,14.344788321991853,0.6707495542934571,0.41990593482401284,44.475727,40.144894
+GDM_g,0.7342980200772142,6.267061316641904,0.7374583145836184,0.26570197992278577,31.813484,29.62764

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/species_metrics.csv

@@ -0,0 +1,7 @@
+Species,Count,R2_Dry_Green_g,Mean_True_Dry_Green_g,R2_Dry_Clover_g,Mean_True_Dry_Clover_g,R2_Dry_Dead_g,Mean_True_Dry_Dead_g,R2_Dry_Total_g,Mean_True_Dry_Total_g,R2_GDM_g,Mean_True_GDM_g
+Lucerne,8,0.5055071115493774,46.076225,0.0,0.0,-0.24506855010986328,8.9362755,0.5930445194244385,55.0125,0.4526757597923279,46.076225
+Ryegrass,23,0.3527984023094177,21.1807,0.0781281590461731,0.44608697,0.07320922613143921,15.000238,0.5457207560539246,36.627026,0.28682899475097656,21.626785
+Phalaris_Clover,17,0.5954955816268921,30.638334,0.2176758050918579,6.0256295,0.19996410608291626,8.100741,0.6213120222091675,44.764706,0.7410099506378174,36.663967
+Clover,16,-5.553554058074951,3.4180498,-0.3775709867477417,23.759737,-0.4449087381362915,4.781,0.6479085683822632,31.95881,0.21597307920455933,27.177788
+Fescue,11,0.6731711626052856,50.04242,0.0,0.0,-0.2130047082901001,27.821217,0.15402120351791382,77.86364,0.6302481889724731,50.04242
+Ryegrass_Clover,30,0.7642334699630737,22.403572,0.39791786670684814,5.9385095,0.16925138235092163,11.830438,0.5706312656402588,40.172512,0.8685345649719238,28.342083

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/state_metrics.csv

@@ -0,0 +1,5 @@
+State,Count,R2_Dry_Green_g,Mean_True_Dry_Green_g,R2_Dry_Clover_g,Mean_True_Dry_Clover_g,R2_Dry_Dead_g,Mean_True_Dry_Dead_g,R2_Dry_Total_g,Mean_True_Dry_Total_g,R2_GDM_g,Mean_True_GDM_g
+NSW,25,0.5327945947647095,49.486637,-0.08623278141021729,0.403924,0.0209619402885437,18.281439,0.45548635721206665,68.172005,0.5050084590911865,49.890556
+Tas,47,0.4930054545402527,16.058243,0.6656339764595032,5.9217763,0.01054239273071289,15.222416,0.4728606343269348,37.202446,0.7668008208274841,21.980019
+Vic,37,0.6630922555923462,26.161087,0.20934492349624634,6.7345862,0.47546929121017456,9.035613,0.7730259895324707,41.931286,0.8399694561958313,32.895668
+WA,10,-1.7467710971832275,3.0211198,-0.4283488988876343,25.812878,0.0,0.0,0.47366201877593994,28.834003,0.12990790605545044,28.834003

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/tail_analysis.csv

@@ -0,0 +1,6 @@
+Target,R2_Bottom10,R2_Middle80,R2_Top10,MAE_Bottom10,MAE_Middle80,MAE_Top10,N_Bottom10,N_Top10
+Dry_Green_g,-425.5793151855469,0.3746812343597412,-0.2598743438720703,7.60463285446167,7.064783573150635,16.724802017211914,11,12
+Dry_Clover_g,,0.5258556604385376,-7.903158187866211,,2.1681559085845947,22.899946212768555,0,12
+Dry_Dead_g,0.0,0.1797838807106018,-4.324334621429443,2.680318593978882,5.405214309692383,30.523239135742188,12,12
+Dry_Total_g,-3.9295778274536133,0.42674243450164795,-3.5205183029174805,7.240514755249023,8.777863502502441,31.870492935180664,11,12
+GDM_g,-2.9886209964752197,0.3634985089302063,-0.7689026594161987,4.068878650665283,6.648159027099609,17.238096237182617,11,12

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/eda/eda_fold_0/variance_analysis.csv

@@ -0,0 +1,6 @@
+Target,True_Var,Pred_Var,Var_Ratio,True_Std,Pred_Std,Std_Ratio
+Dry_Green_g,504.8085,391.8102,0.7761561,22.467943,19.794197,0.8809973
+Dry_Clover_g,136.71416,30.455727,0.22276936,11.692483,5.5186706,0.47198448
+Dry_Dead_g,195.24431,40.33029,0.20656322,13.972985,6.3506136,0.45449224
+Dry_Total_g,771.6048,387.10733,0.50169116,27.777775,19.675043,0.7083016
+GDM_g,494.254,361.37445,0.7311513,22.231825,19.009851,0.8550738

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/history.json

@@ -0,0 +1,515 @@
+{
+  "train_loss": [
+    38.25010136195591,
+    36.53139236995152,
+    31.988073348999023,
+    24.577355248587473,
+    19.159065008163452,
+    16.32198647090367,
+    14.3743873664311,
+    12.156169891357422,
+    10.86801188332694,
+    11.426748139517647,
+    10.776491062981743,
+    10.70757566179548,
+    9.42965565408979,
+    9.098866973604474,
+    8.745556456702095,
+    8.672296830586024,
+    8.088422502790179,
+    8.225612231663295,
+    7.931447471891131,
+    7.4458067417144775,
+    7.516281877245222,
+    7.327921560832432,
+    7.377258437020438,
+    7.159599099840436,
+    7.168285710471017
+  ],
+  "val_loss": [
+    37.62777769265055,
+    33.88804458169376,
+    27.30264609200614,
+    20.47000009873334,
+    19.796937205210455,
+    14.595191138131279,
+    13.924036675140638,
+    12.628718071625013,
+    12.876988875765761,
+    12.018007102132845,
+    11.97453432323552,
+    10.936345845711331,
+    11.081135084649095,
+    10.882176311076188,
+    11.307619215059681,
+    10.60511861528669,
+    9.91043189393372,
+    10.303135118564638,
+    9.870684944281058,
+    9.636943969406,
+    9.707796914236885,
+    9.55041787604324,
+    9.53952128947282,
+    9.530494946391643,
+    9.530849168280593
+  ],
+  "train_r2": [
+    -1.3363992428301796,
+    -1.3196537772666352,
+    -0.8873478758391011,
+    -0.20886045367830586,
+    0.24811554947129089,
+    0.49291333858822617,
+    0.5562232002036308,
+    0.6319377353486892,
+    0.7338443355094496,
+    0.692102891841907,
+    0.7244905414275532,
+    0.7123194831546062,
+    0.7836211120436584,
+    0.783140524718787,
+    0.7896748664128698,
+    0.7940734773980354,
+    0.8325203733205339,
+    0.8197104027192132,
+    0.8291330133884642,
+    0.8411863445885773,
+    0.8408465787217054,
+    0.8512713272457689,
+    0.8409852882007028,
+    0.8591461457571269,
+    0.8649198419986462
+  ],
+  "val_r2": [
+    -1.3950973985146478,
+    -1.198703639298798,
+    -0.5138442783813648,
+    0.137136798623971,
+    0.40409173418679656,
+    0.5107153135892943,
+    0.5658018314025428,
+    0.6295323394142928,
+    0.5831503926006005,
+    0.6605273238128424,
+    0.6020487875682785,
+    0.6836109195362274,
+    0.6976443925987601,
+    0.6990410879133884,
+    0.7042520547042326,
+    0.7096293930351047,
+    0.7144549421067079,
+    0.6766666200690996,
+    0.7253673436920736,
+    0.7205189667092756,
+    0.7309251869336233,
+    0.7255881999084609,
+    0.7229875670343322,
+    0.7261219338988341,
+    0.7266583495616307
+  ],
+  "train_legacy_r2": [
+    -1.799535346031189,
+    -1.797531497478485,
+    -1.2777204751968383,
+    -0.5116644859313965,
+    -0.004630404710769653,
+    0.26667945384979247,
+    0.3424946486949921,
+    0.43333588242530824,
+    0.5465874969959259,
+    0.5003551125526429,
+    0.5468904435634613,
+    0.5324286937713623,
+    0.6335276961326599,
+    0.6323913216590882,
+    0.6443242728710175,
+    0.6554654479026795,
+    0.706923907995224,
+    0.6960991561412812,
+    0.7055196642875672,
+    0.7184698820114136,
+    0.7139967262744904,
+    0.7306373834609985,
+    0.7268960118293762,
+    0.737014251947403,
+    0.7464584171772004
+  ],
+  "val_legacy_r2": [
+    -1.870447814464569,
+    -1.64388747215271,
+    -0.8639281749725342,
+    -0.13160667419433594,
+    0.17489959001541136,
+    0.30283214449882506,
+    0.3659003674983978,
+    0.44055628776550293,
+    0.3881571888923645,
+    0.4778317749500275,
+    0.42730675935745244,
+    0.5162340462207794,
+    0.5384969294071198,
+    0.5451394140720368,
+    0.5503804445266723,
+    0.5659677624702454,
+    0.5756272971630096,
+    0.5368889808654785,
+    0.5909524083137513,
+    0.5887383937835693,
+    0.6003269851207733,
+    0.59625643491745,
+    0.5938564419746399,
+    0.5971087217330933,
+    0.5976428031921387
+  ],
+  "train_mean_r2": [
+    -1.273852825164795,
+    -1.2843811750411986,
+    -0.9017316341400147,
+    -0.36063945293426514,
+    -0.008813130855560302,
+    0.18296027183532715,
+    0.2520625710487366,
+    0.3361587166786194,
+    0.43007329702377317,
+    0.3823347449302673,
+    0.43260403871536257,
+    0.430102801322937,
+    0.529281485080719,
+    0.5281005144119263,
+    0.5483026504516602,
+    0.5692844152450561,
+    0.6171696901321411,
+    0.6123136878013611,
+    0.616560447216034,
+    0.6257499575614929,
+    0.6181848168373107,
+    0.6373738169670105,
+    0.6476767897605896,
+    0.6389514803886414,
+    0.6473283529281616
+  ],
+  "val_mean_r2": [
+    -1.3570065259933473,
+    -1.192777681350708,
+    -0.6283692359924317,
+    -0.09932615756988525,
+    0.12574578523635865,
+    0.24423736333847046,
+    0.29086569547653196,
+    0.35089247226715087,
+    0.3097109913825989,
+    0.3790387034416199,
+    0.3696777105331421,
+    0.4263035416603088,
+    0.4524629235267639,
+    0.46540423631668093,
+    0.4638331413269043,
+    0.4962760329246521,
+    0.5134007811546326,
+    0.4942257642745972,
+    0.5274017333984375,
+    0.532895851135254,
+    0.539402163028717,
+    0.5411573886871338,
+    0.5409189701080322,
+    0.5423749923706055,
+    0.5425600051879883
+  ],
+  "lr": [
+    1.3864160009206661e-05,
+    3.61219307637806e-05,
+    3.995128295395628e-05,
+    3.956296949659553e-05,
+    3.879390066162154e-05,
+    3.7659045499104253e-05,
+    3.618049267390345e-05,
+    3.438702053493275e-05,
+    3.2313536977332906e-05,
+    3.0000400000000005e-05,
+    2.7492632183043514e-05,
+    2.4839044374476875e-05,
+    2.209128564258246e-05,
+    1.9302837985547353e-05,
+    1.652797536520353e-05,
+    1.3820707326096555e-05,
+    1.1233727761135883e-05,
+    8.817389284907844e-06,
+    6.61872317730793e-06,
+    4.680523973174938e-06,
+    3.0405165153484065e-06,
+    1.7306216835140948e-06,
+    7.763350905903729e-07,
+    1.9623083962358603e-07,
+    1.6e-09
+  ],
+  "train_Dry_Green_g": [
+    -0.958661675453186,
+    -1.0082142353057861,
+    -0.750934362411499,
+    -0.3271596431732178,
+    0.14792495965957642,
+    0.3978475332260132,
+    0.4779917597770691,
+    0.6113640666007996,
+    0.7039214968681335,
+    0.6091651320457458,
+    0.6886366605758667,
+    0.6869487166404724,
+    0.7330451607704163,
+    0.7363078594207764,
+    0.7521035075187683,
+    0.7778913378715515,
+    0.8013703227043152,
+    0.7688601613044739,
+    0.7801900506019592,
+    0.8067132830619812,
+    0.8148974180221558,
+    0.8075028657913208,
+    0.8076215982437134,
+    0.8085763454437256,
+    0.8245816826820374
+  ],
+  "val_Dry_Green_g": [
+    -1.1841833591461182,
+    -1.0881059169769287,
+    -0.6200168132781982,
+    0.00666886568069458,
+    0.21917808055877686,
+    0.48407816886901855,
+    0.5464293956756592,
+    0.6299786567687988,
+    0.5456879138946533,
+    0.6244778633117676,
+    0.6175287961959839,
+    0.6444178819656372,
+    0.6502053737640381,
+    0.6580924987792969,
+    0.6041362285614014,
+    0.670886218547821,
+    0.7147735953330994,
+    0.6981797814369202,
+    0.7021872997283936,
+    0.7143092155456543,
+    0.7076495289802551,
+    0.7176140546798706,
+    0.7203965187072754,
+    0.7188331484794617,
+    0.7184714078903198
+  ],
+  "train_Dry_Clover_g": [
+    -0.28491127490997314,
+    -0.27291882038116455,
+    -0.1868687868118286,
+    -0.06844913959503174,
+    -0.12927758693695068,
+    -0.11837387084960938,
+    -0.06297957897186279,
+    -0.0391237735748291,
+    0.006886601448059082,
+    0.003437817096710205,
+    0.04300403594970703,
+    0.07734721899032593,
+    0.18083852529525757,
+    0.20603108406066895,
+    0.24135881662368774,
+    0.32740074396133423,
+    0.3150603175163269,
+    0.3713289499282837,
+    0.38948720693588257,
+    0.370074987411499,
+    0.3585164546966553,
+    0.34358668327331543,
+    0.4318656325340271,
+    0.41063743829727173,
+    0.3234837055206299
+  ],
+  "val_Dry_Clover_g": [
+    -0.32436859607696533,
+    -0.23591089248657227,
+    -0.12310922145843506,
+    -0.1636648178100586,
+    -0.08971655368804932,
+    -0.07593047618865967,
+    -0.06065833568572998,
+    -0.022351861000061035,
+    -0.01968526840209961,
+    0.0059812068939208984,
+    0.16423308849334717,
+    0.14826107025146484,
+    0.2327316403388977,
+    0.2666144371032715,
+    0.3073257803916931,
+    0.3376161456108093,
+    0.37975865602493286,
+    0.39284205436706543,
+    0.42185765504837036,
+    0.4275532364845276,
+    0.4370489716529846,
+    0.4425346255302429,
+    0.4423545002937317,
+    0.4424399137496948,
+    0.4424739480018616
+  ],
+  "train_Dry_Dead_g": [
+    -1.035466194152832,
+    -1.0517761707305908,
+    -0.6200674772262573,
+    -0.19200313091278076,
+    -0.0779261589050293,
+    -0.04009723663330078,
+    -0.04310107231140137,
+    0.023473143577575684,
+    0.058447957038879395,
+    0.02960991859436035,
+    0.07235842943191528,
+    0.029353678226470947,
+    0.21829700469970703,
+    0.16862010955810547,
+    0.21078580617904663,
+    0.19486373662948608,
+    0.3334410786628723,
+    0.3038480281829834,
+    0.2692434787750244,
+    0.2766001224517822,
+    0.25150638818740845,
+    0.3310643434524536,
+    0.3282850384712219,
+    0.2602514624595642,
+    0.36153924465179443
+  ],
+  "val_Dry_Dead_g": [
+    -0.8277060985565186,
+    -0.730534553527832,
+    -0.2945929765701294,
+    0.007916510105133057,
+    0.06966942548751831,
+    0.06982666254043579,
+    0.04473376274108887,
+    0.04252815246582031,
+    0.050161898136138916,
+    0.08253198862075806,
+    -0.0002416372299194336,
+    0.08627301454544067,
+    0.09876912832260132,
+    0.09584760665893555,
+    0.11473524570465088,
+    0.1553095579147339,
+    0.14262431859970093,
+    0.16651791334152222,
+    0.15381211042404175,
+    0.1767657995223999,
+    0.17506885528564453,
+    0.18478333950042725,
+    0.1907835602760315,
+    0.19046515226364136,
+    0.19017571210861206
+  ],
+  "train_Dry_Total_g": [
+    -2.5119547843933105,
+    -2.4881374835968018,
+    -1.7725863456726074,
+    -0.6992874145507812,
+    -0.0058171749114990234,
+    0.35885608196258545,
+    0.4254106283187866,
+    0.5224950313568115,
+    0.6447993516921997,
+    0.6074721813201904,
+    0.6489543914794922,
+    0.6056361794471741,
+    0.7248809337615967,
+    0.7179557085037231,
+    0.7214881181716919,
+    0.7206553816795349,
+    0.7824712991714478,
+    0.7606306076049805,
+    0.7761709690093994,
+    0.793529212474823,
+    0.7943464517593384,
+    0.8049298524856567,
+    0.7866548299789429,
+    0.8200309872627258,
+    0.8336350917816162
+  ],
+  "val_Dry_Total_g": [
+    -2.4902236461639404,
+    -2.188549757003784,
+    -1.1311028003692627,
+    -0.1572948694229126,
+    0.23022300004959106,
+    0.3546407222747803,
+    0.42695051431655884,
+    0.5155977010726929,
+    0.45354217290878296,
+    0.5669739246368408,
+    0.4526035785675049,
+    0.5927523374557495,
+    0.6140154004096985,
+    0.6059687733650208,
+    0.6305567622184753,
+    0.6202430725097656,
+    0.619807243347168,
+    0.5613906383514404,
+    0.637787938117981,
+    0.6256845593452454,
+    0.6430051922798157,
+    0.6319741606712341,
+    0.6276364326477051,
+    0.6330251693725586,
+    0.6339830160140991
+  ],
+  "train_GDM_g": [
+    -1.5782701969146729,
+    -1.6008591651916504,
+    -1.1782011985778809,
+    -0.5162979364395142,
+    0.021030306816101074,
+    0.31656885147094727,
+    0.4629911184310913,
+    0.5625851154327393,
+    0.7363110780715942,
+    0.6619886755943298,
+    0.7100666761398315,
+    0.7512282133102417,
+    0.7893458008766174,
+    0.8115878105163574,
+    0.8157770037651062,
+    0.825610876083374,
+    0.8535054326057434,
+    0.856900691986084,
+    0.867710530757904,
+    0.8818321824073792,
+    0.8716573715209961,
+    0.8997853398323059,
+    0.8839568495750427,
+    0.8952611684799194,
+    0.8934020400047302
+  ],
+  "val_GDM_g": [
+    -1.9585509300231934,
+    -1.7207872867584229,
+    -0.9730243682861328,
+    -0.19025647640228271,
+    0.1993749737739563,
+    0.38857173919677734,
+    0.496873140335083,
+    0.5887097120285034,
+    0.5188482403755188,
+    0.615228533744812,
+    0.614264726638794,
+    0.659813404083252,
+    0.666593074798584,
+    0.7004978656768799,
+    0.6624116897583008,
+    0.6973251700401306,
+    0.7100400924682617,
+    0.6521984338760376,
+    0.7213636636734009,
+    0.7201664447784424,
+    0.7342382669448853,
+    0.728880763053894,
+    0.7234238386154175,
+    0.7271115779876709,
+    0.7276959419250488
+  ]
+}

day6/convnextv2l_nobias_huber_2grid_film_12heads_derivev2_f33_aux_convnextv2_large_20251228_2142/fold_0/results.json

@@ -0,0 +1,8 @@
+{
+  "best_r2": 0.7309251869336233,
+  "best_epoch": 21,
+  "final_train_r2": 0.8649198419986462,
+  "final_val_r2": 0.7266583495616307,
+  "final_train_legacy_r2": 0.7464584171772004,
+  "final_val_legacy_r2": 0.5976428031921387
+}

day6/dinov2b_huber_4grid_16heads_bottom_up_vit_base_patch14_reg4_dinov2_20251228_2203/config.json

@@ -0,0 +1,118 @@
+{
+  "name": "dinov2b_huber_4grid_16heads_bottom_up",
+  "description": "dinov2-base with Huber loss",
+  "tags": [
+    "baseline",
+    "dinov2"
+  ],
+  "paths": {
+    "data_dir": "/csiro-biomass_data/",
+    "train_csv": "train.csv",
+    "test_csv": "test.csv",
+    "train_images": "train",
+    "test_images": "test",
+    "output_dir": "/volume/exp/day6"
+  },
+  "model": {
+    "backbone": "timm/vit_base_patch14_reg4_dinov2.lvd142m",
+    "pretrained": true,
+    "freeze_percent": 0.66,
+    "img_size": 518,
+    "grid": [
+      4,
+      4
+    ],
+    "use_stereo": true,
+    "stereo_heads": 16,
+    "dropout": 0.35,
+    "use_global": true,
+    "global_weight_learnable": true,
+    "head_strategy": "derive",
+    "use_dead_ratio_head": true,
+    "use_film": true,
+    "film_hidden_dim": null,
+    "film_hidden_ratio": 0.4,
+    "film_dropout": 0.1,
+    "film_share": false,
+    "film_apply_to_global": false,
+    "use_image_features": false
+  },
+  "data": {
+    "batch_size": 5,
+    "num_workers": 4,
+    "pin_memory": true,
+    "use_augmentation": true,
+    "aug_hflip": 0.3,
+    "aug_vflip": 0.3,
+    "aug_rotate90": 0.3,
+    "aug_brightness_contrast": 0.2,
+    "aug_hue_sat": 0.0,
+    "aug_noise": 0.1
+  },
+  "fold": {
+    "n_folds": 3,
+    "fold_idx": null,
+    "stratify": true,
+    "stratify_column": "Dry_Total_g",
+    "stratify_columns": [
+      "State",
+      "Dry_Total_g"
+    ],
+    "stratify_bins": 5,
+    "include_month": true,
+    "shuffle": true,
+    "test_size": 0.2
+  },
+  "loss": {
+    "loss_type": "huber",
+    "huber_delta": 1.0,
+    "weight_total": 0.3,
+    "weight_gdm": 0.25,
+    "weight_green": 0.2,
+    "weight_clover": 0.2,
+    "weight_dead": 0.2
+  },
+  "optimizer": {
+    "optimizer": "adamw",
+    "learning_rate": "1e-5",
+    "weight_decay": 0.01,
+    "scheduler": "onecycle",
+    "warmup_epochs": 1,
+    "warmup_pct": 0.1,
+    "min_lr": "4e-7",
+    "gradient_accumulation_steps": 1,
+    "max_grad_norm": 0.0
+  },
+  "training": {
+    "epochs": 26,
+    "seed": 42,
+    "mixed_precision": true,
+    "patience": 7,
+    "early_stopping_metric": "val_weighted_r2",
+    "early_stopping_mode": "max",
+    "save_best_only": true,
+    "save_last": true
+  },
+  "aux": {
+    "enabled": true,
+    "predict_height": true,
+    "height_target": "norm",
+    "height_weight": 0.05,
+    "predict_ndvi": true,
+    "ndvi_target": "norm",
+    "ndvi_weight": 0.05,
+    "predict_state": true,
+    "state_weight": 0.035,
+    "state_num_classes": 4,
+    "state_class_weighting": "balanced",
+    "state_class_weight_max": 10.0,
+    "state_effective_num_beta": 0.999,
+    "predict_species": false,
+    "species_weight": 0.06,
+    "species_num_classes": null,
+    "species_class_weighting": "effective_num",
+    "species_class_weight_max": 10.0,
+    "species_effective_num_beta": 0.999,
+    "label_smoothing": 0.015
+  }
+}