Add training metadata and checksums: geometry_ratio/depth10_normals20

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/COMPLETE.json

@@ -0,0 +1,7 @@
+{
+  "status": "COMPLETE",
+  "attempt": "1",
+  "completed_utc": "2026-08-12T09:42:36.751883+00:00",
+  "completion_source": "trainer_exit_zero",
+  "wrapper_exit_code": 0
+}

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/SHA256SUMS

@@ -0,0 +1,6 @@
+be895c2602b3d07fbbe7abb8ef7322315c4d5aab5db822a4fdc5f56ee91db42a  config.json
+235a30e4d3ac93b6eaf483fe690af3d302d18b9e883878695fa60aff5ff0818b  diffusion_pytorch_model.safetensors.index.json
+242f5575ae52c98692cf6b7b5a8eb9b47e4064d9476d52e3a59f645105ae1775  diffusion_pytorch_model-00001-of-00004.safetensors
+020d5bbd08c2c7bf0c16ed140f1a45da0c17cb50383a4a241d0b0d4e631d5e44  diffusion_pytorch_model-00002-of-00004.safetensors
+f24a7331cd8fea4c5c9fa1a3f93c2159266e40935cd46186a469bc6b17e79467  diffusion_pytorch_model-00003-of-00004.safetensors
+772b7a06e1b5cf85b6027a54a5439b26284cc1149aa829c72bb02e07c3c2de57  diffusion_pytorch_model-00004-of-00004.safetensors

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/checksum_validation.json

@@ -0,0 +1,30 @@
+{
+  "fingerprints": {
+    "config.json": {
+      "size": 563,
+      "mtime_ns": 1786527721000000000
+    },
+    "diffusion_pytorch_model.safetensors.index.json": {
+      "size": 24875,
+      "mtime_ns": 1786527750000000000
+    },
+    "diffusion_pytorch_model-00001-of-00004.safetensors": {
+      "size": 9804206048,
+      "mtime_ns": 1786527729000000000
+    },
+    "diffusion_pytorch_model-00002-of-00004.safetensors": {
+      "size": 9797913704,
+      "mtime_ns": 1786527736000000000
+    },
+    "diffusion_pytorch_model-00003-of-00004.safetensors": {
+      "size": 9596584288,
+      "mtime_ns": 1786527744000000000
+    },
+    "diffusion_pytorch_model-00004-of-00004.safetensors": {
+      "size": 7115649136,
+      "mtime_ns": 1786527750000000000
+    }
+  },
+  "sha256_manifest": "6f638635a6c80d0cb359791c0b27ca1ea6f9d6a01832c4eb991933586bb1a1ff",
+  "validation": "PASS"
+}

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/dataset_specs.json

@@ -0,0 +1,282 @@
+{
+  "datasets": [
+    {
+      "filters": {
+        "source_dataset": "acdc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_acdc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "bdd"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_bdd",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "voc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.032
+    },
+    {
+      "filters": {
+        "source_dataset": "mapillary"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_mapillary",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.032
+    },
+    {
+      "filters": {
+        "source_dataset": "voc_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.016
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "refseg",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.08
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "pico_banana",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.15
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "identity",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.1
+    },
+    {
+      "name": "pd12m",
+      "shards": "<local-path>",
+      "source_dataset": "pd12m",
+      "streaming": {
+        "error_policy": "warn_and_continue",
+        "resampled": true,
+        "sample_shuffle_buffer": 10000,
+        "shard_shuffle": true
+      },
+      "task": "text_to_image",
+      "type": "webdataset_text_to_image",
+      "weight": 0.15
+    },
+    {
+      "archive_sha256": "36aac18f3ff4a51226def096a8f9ffd3bb4ee8f86ccda9e78f19ba50c407c344",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "ceed5191233ce3b4bf1dd14fa869ca0f935bd0604f9a3ffd9d6f1d3ff326d632",
+      "name": "depth_hypersim",
+      "path": "<local-path>",
+      "sample_list_sha256": "f7b774dbf2645993955fbd315e2a0edc7f499ba3a9513a4101f481dc7a1a9cf7",
+      "sampling": "scene_balanced",
+      "source_dataset": "hypersim",
+      "task": "depth_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.02
+    },
+    {
+      "archive_sha256": "8968b57be59dc69a66d5cffaff7667e6836188e7c849003bdf66e5e4ff01b18a",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "ac4f12e2ddcb33ee53425ccc085d4c2ac07a2c776e0e9b20f3bbf371ae636e55",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sample_list_sha256": "37280ca662f240992afc5fc0400f1a5d27d4bbb08f61665bd46ac3d27cb0e5ff",
+      "sampling": "scene_balanced",
+      "source_dataset": "matterport3d",
+      "task": "depth_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.02
+    },
+    {
+      "archive_sha256": "0840146fd580149b35cf67207a176812c207851a0880b0dbfbe5b479c1417246",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "327b3e2cbec5092f6606ee551f9387b524f5493693ed1bc9ee4353ceca1268d9",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sample_list_sha256": "89bcff6ca51ff82624ff44aa986bf6982fe7863195f63d4865e889c8560700bb",
+      "sampling": "scene_balanced",
+      "source_dataset": "replica",
+      "task": "depth_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.02
+    },
+    {
+      "archive_sha256": "7d62865ca5768e5888b88d9fab9e9de9d6a44cf796e3a22d86105b6c3d0e1d33",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "381e6a1454f6aceae4030ca2bbb23a6f855e1608a3bc6213b2b14c675c3c3f7f",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sample_list_sha256": "7118294ad359f2cb9c1acb6a93719a3a6769f3dbaa4b646a216f8788de961520",
+      "sampling": "row",
+      "source_dataset": "synscapes",
+      "task": "depth_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.02
+    },
+    {
+      "archive_sha256": "179cec3d7309089a9c16c41a6ec2a8f6339d30e96feb633453f113793507963b",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "2821d90b4432419169d7f98cbd515bffc2069cc6016a317c2c23c260d0e2eaff",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sample_list_sha256": "698aad7a87803c74ff750bfaa2cd37fbdf29d4b5b73e77009aea4b41c424c7d2",
+      "sampling": "scene_balanced",
+      "source_dataset": "taskonomy",
+      "task": "depth_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.02
+    },
+    {
+      "archive_sha256": "40a0424cfb4484ea5edf7ca773ecbc52d5cb603240a4bc6e12afa085d0db0fba",
+      "geometry_version": "geometry_mixed_v2_v3",
+      "index_path": "<local-path>",
+      "index_sha256": "e4346e4a727ba3e1cdc58391368c51025e242d54b2b8f69cc795162e1a3e347c",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sample_list_sha256": "34d6036cea27350c8cbba0a51614388561ce08b89871eafd0094c3b758572fd0",
+      "sampling": "scene_balanced",
+      "source_dataset": "hypersim",
+      "task": "surface_normal_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.04
+    },
+    {
+      "archive_sha256": "735630652f130fd7209b5f53ca5343d141935e15166fdf939db14f1996b891d5",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "1c19051fdd1c248ce986f8c231d7ac0714695dd45ae68cbee506d602700548af",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sample_list_sha256": "37280ca662f240992afc5fc0400f1a5d27d4bbb08f61665bd46ac3d27cb0e5ff",
+      "sampling": "scene_balanced",
+      "source_dataset": "matterport3d",
+      "task": "surface_normal_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.04
+    },
+    {
+      "archive_sha256": "bb0c1653e5802890d2cc7358941750d9020d38081b9d00882d84ec832efed501",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "082e88258e104d1e8c96e1abaef8991506cd2ce4951b1c402ad6fab439ed5feb",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sample_list_sha256": "89bcff6ca51ff82624ff44aa986bf6982fe7863195f63d4865e889c8560700bb",
+      "sampling": "scene_balanced",
+      "source_dataset": "replica",
+      "task": "surface_normal_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.04
+    },
+    {
+      "archive_sha256": "d4e70dbeb5ef281be17a304ab89f7253cec8d2ba716d307e855c788d57b125e3",
+      "geometry_version": "geometry_canonical_v2",
+      "index_path": "<local-path>",
+      "index_sha256": "10d2f4bbf5e698f60657f9e0e06eb6d1d6319939d2569d89983503981dfbb93e",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sample_list_sha256": "7118294ad359f2cb9c1acb6a93719a3a6769f3dbaa4b646a216f8788de961520",
+      "sampling": "row",
+      "source_dataset": "synscapes",
+      "task": "surface_normal_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.04
+    },
+    {
+      "archive_sha256": "a3550f40fcfe60a72622cd3f1cef38f5eb4870c9653a50a740a3c679d31accb4",
+      "geometry_version": "geometry_vision_banana_v3",
+      "index_path": "<local-path>",
+      "index_sha256": "0f01831b926d04216e173624146a693e96d95728a9bd234a974313400d97bae5",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sample_list_sha256": "698aad7a87803c74ff750bfaa2cd37fbdf29d4b5b73e77009aea4b41c424c7d2",
+      "sampling": "scene_balanced",
+      "source_dataset": "taskonomy",
+      "task": "surface_normal_estimation",
+      "type": "indexed_prepared_tar",
+      "validation_report": "<local-path>",
+      "weight": 0.04
+    }
+  ],
+  "excluded_geometry_sources": [],
+  "experiment_id": "e8_ratio_depth10_normals20",
+  "profile": "geometry_ablation_20260801_v1",
+  "samples_per_epoch": 160000,
+  "schema_version": 2,
+  "seed": 42,
+  "shuffle": true,
+  "source_weighting": "equal_within_geometry_task_after_exclusion",
+  "task_group_weights": {
+    "depth": 0.1,
+    "non_geometry": 0.7,
+    "surface_normals": 0.2
+  }
+}

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/dataset_specs.resolved.json

@@ -0,0 +1,202 @@
+{
+  "datasets": [
+    {
+      "filters": {
+        "source_dataset": "acdc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_acdc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "bdd"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_bdd",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "voc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.035
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.032
+    },
+    {
+      "filters": {
+        "source_dataset": "mapillary"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_mapillary",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.032
+    },
+    {
+      "filters": {
+        "source_dataset": "voc_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.016
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "refseg",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.08
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "pico_banana",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.15
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "identity",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.1
+    },
+    {
+      "name": "pd12m",
+      "shards": "<local-path>",
+      "source_dataset": "pd12m",
+      "streaming": {
+        "error_policy": "warn_and_continue",
+        "resampled": true,
+        "sample_shuffle_buffer": 10000,
+        "shard_shuffle": true
+      },
+      "task": "text_to_image",
+      "type": "webdataset_text_to_image",
+      "weight": 0.15
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    }
+  ],
+  "samples_per_epoch": 160000,
+  "seed": 42,
+  "shuffle": true
+}

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/evaluation_protocol.json

@@ -0,0 +1,7 @@
+{
+  "scope": "Trained ablation checkpoint; consult the cited study for datasets and inference settings.",
+  "corrected_depth": "Marigold-v2 PPD per-image log-depth RANSAC alignment",
+  "corrected_normals": "geometry-v6 assets and fixed camera-facing decoder; no test-set axis fitting",
+  "historical_results": "Historical tables use different assets, datasets or scoring conventions; do not mix with corrected scores.",
+  "checkpoint_selection": "Evaluated checkpoint; intermediate training-duration and warmup checkpoints are retained."
+}

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/run_spec.json

@@ -0,0 +1,91 @@
+{
+  "account": "h200_y506dd10",
+  "campaign_id": "geometry_klein_37500_continuations_20260811_v1",
+  "continuation": {
+    "mode": "exact_optimizer_scheduler_rng_sampler_resume",
+    "source_checkpoint": "<local-path>",
+    "source_checkpoint_step": 20000,
+    "source_dataset_specs_sha256": "548b407be93526bd1aa1bd29385668db39e577bd9e217eb351d5bd37a3e842ca",
+    "source_run": "<local-path>",
+    "source_run_spec_sha256": "95de623c86ba93259354882ac0ca5c230f3311ae76881d6d75fa7bebd1100df5",
+    "target_checkpoint_step": 37500
+  },
+  "environment": {
+    "ALLOW_LEGACY_HYPERSIM_RANGE_DEPTH": "0",
+    "ALLOW_UNVERIFIED_NORMAL_SOURCES": "0",
+    "ASPECT_RATIO_BUCKETS": "512,512;576,448;448,576;608,416;416,608",
+    "AUTO_RECONCILE_ON_COMPLETE": "0",
+    "CACHE_LATENTS": "0",
+    "CHECKPOINTING_STEPS": "500",
+    "CHECKPOINTS_TOTAL_LIMIT": "3",
+    "CUDA_VISIBLE_DEVICES": "0,1,2,3",
+    "DATASET_SPECS": "<local-path>",
+    "DEPTH_ENCODING": "relative_inverse_grayscale",
+    "DEPTH_PROMPT_CONTRACT": "campaign_detailed_inverse",
+    "DEPTH_TARGET_RESIZE_POLICY": "metric_field_then_encode",
+    "DISTRIBUTED_STRATEGY": "fsdp",
+    "FSDP_TRANSFORMER_DTYPE_POLICY": "fp32_all",
+    "GEOMETRY_TASKS": "depth,normals",
+    "GRADIENT_ACCUMULATION_STEPS": "2",
+    "GUIDANCE_SCALE": "3.5",
+    "LEARNING_RATE": "2e-06",
+    "LOSS_MASK_EROSION_RADIUS": "0",
+    "LOSS_MASK_MIN_VALID_FRACTION": "0.5",
+    "LOSS_MASK_MODE": "hard",
+    "LOSS_MASK_POLICY": "auto",
+    "LOSS_MASK_REDUCTION": "per_sample",
+    "LR_SCHEDULER": "constant_with_warmup",
+    "LR_WARMUP_STEPS": "1000",
+    "MAX_TRAIN_STEPS": "37500",
+    "MIXED_PRECISION": "bf16",
+    "MODEL": "<local-path>",
+    "MODEL_FAMILY": "flux2_klein",
+    "NORMAL_PROMPT_CONTRACT": "campaign_short",
+    "NORMAL_TARGET_ENCODING": "vision_banana_x_left_red",
+    "NORMAL_TARGET_RESIZE_POLICY": "vector_field_renormalize_then_encode",
+    "NUM_PROCESSES": "4",
+    "OUTPUT_DIR": "<local-path>",
+    "PERMANENT_CHECKPOINT_STEPS": "20000,37500",
+    "PROMPT_ENCODING_MODE": "auto",
+    "REPORT_TO": "tensorboard",
+    "RESAMPLE_DATASET_EACH_EPOCH": "1",
+    "RESOLUTION": "512",
+    "RESUME_FROM_CHECKPOINT": "latest",
+    "ROOT": "<local-path>",
+    "RUN_DIR": "<local-path>",
+    "RUN_ID": "c375_ratio_depth10_normals20",
+    "SAMPLES_PER_EPOCH": "160000",
+    "SEED": "42",
+    "SKIP_FINAL_INFERENCE": "1",
+    "SKIP_FINAL_SAVE": "1",
+    "TASK_HOMOGENEOUS_BATCHES": "1",
+    "TRAINING_SUITE": "geometry_ablation_20260801_v1",
+    "TRAIN_ALL_TRANSFORMER_BLOCKS": "1",
+    "TRAIN_BATCH_SIZE": "1",
+    "TRAIN_FULL_TRANSFORMER": "0",
+    "TRAIN_INPUT_LAYERS": "0",
+    "TRAIN_LAST_SINGLE_TRANSFORMER_BLOCKS": "0",
+    "TRAIN_LAST_TRANSFORMER_BLOCKS": "0",
+    "TRAIN_METHOD": "partial_full_rank",
+    "TRAIN_NON_BLOCK_LAYERS": "1",
+    "TRAIN_OUTPUT_LAYERS": "1",
+    "WEIGHT_DECAY": "1e-4"
+  },
+  "evaluate_checkpoints": [
+    37500
+  ],
+  "evaluation_scope": "geometry_and_compact_guardrails",
+  "evaluation_suites": [
+    "geometry",
+    "semantic",
+    "refseg",
+    "instance",
+    "imgedit"
+  ],
+  "group": "downstream",
+  "run_id": "c375_ratio_depth10_normals20",
+  "schema_version": 1,
+  "source_campaign_id": "geometry_ablation_20260801_v1",
+  "source_run_id": "e8_ratio_depth10_normals20",
+  "status": "PLANNED"
+}

ablation_models/additional_studies/geometry/depth_normal_allocation/depth10_normals20/checkpoint-37500/training_args.json

@@ -0,0 +1,129 @@
+{
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "adam_epsilon": 1e-08,
+  "adam_weight_decay": 0.0001,
+  "adam_weight_decay_text_encoder": 0.001,
+  "allow_large_prompt_precompute_cache": false,
+  "allow_legacy_hypersim_range_depth": false,
+  "allow_tf32": false,
+  "allow_unverified_normal_sources": false,
+  "angular_normal_aux_weight": 0.0,
+  "aspect_ratio_buckets": "512,512;576,448;448,576;608,416;416,608",
+  "bnb_quantization_config_path": null,
+  "cache_dir": null,
+  "cache_latents": false,
+  "caption_column": "text",
+  "center_crop": false,
+  "checkpointing_steps": 500,
+  "checkpoints_total_limit": 3,
+  "class_data_dir": null,
+  "cond_image_column": "conditioning_image",
+  "dataloader_num_workers": 0,
+  "dataset_config_name": null,
+  "dataset_index_num_workers": 16,
+  "dataset_name": null,
+  "dataset_pool_seed": 42,
+  "dataset_specs": "<local-path>",
+  "depth_prompt_contract": "campaign_detailed_inverse",
+  "depth_target_encoding": "relative_inverse_grayscale",
+  "depth_target_resize_policy": "metric_field_then_encode",
+  "do_fp8_training": false,
+  "enable_npu_flash_attention": false,
+  "final_validation_prompt": null,
+  "fsdp_text_encoder": false,
+  "fsdp_transformer": true,
+  "fsdp_transformer_cls_to_wrap": "Flux2TransformerBlock,Flux2SingleTransformerBlock",
+  "fsdp_transformer_dtype_policy": "fp32_all",
+  "gradient_accumulation_steps": 2,
+  "gradient_checkpointing": true,
+  "guidance_scale": 3.5,
+  "hub_model_id": null,
+  "hub_token": null,
+  "image_column": "image",
+  "instance_data_dir": null,
+  "instance_prompt": "[TASK: IMAGE_EDITING] Edit the image according to the instruction.",
+  "learning_rate": 2e-06,
+  "local_rank": 0,
+  "logging_dir": "logs",
+  "logit_mean": 0.0,
+  "logit_std": 1.0,
+  "lora_alpha": 64,
+  "lora_dropout": 0.0,
+  "lora_layers": null,
+  "lora_rank": 64,
+  "loss_mask_erosion_radius": 0,
+  "loss_mask_min_sample_valid_fraction": 0.0,
+  "loss_mask_min_valid_fraction": 0.5,
+  "loss_mask_mode": "hard",
+  "loss_mask_policy": "auto",
+  "loss_mask_reduction": "per_sample",
+  "lr_num_cycles": 1,
+  "lr_power": 1.0,
+  "lr_scheduler": "constant_with_warmup",
+  "lr_warmup_steps": 1000,
+  "max_grad_norm": 1.0,
+  "max_sequence_length": 512,
+  "max_train_steps": 37500,
+  "metric_depth_aux_max_depth": 80.0,
+  "metric_depth_aux_weight": 0.0,
+  "mixed_precision": "bf16",
+  "mode_scale": 1.29,
+  "normal_prompt_contract": "campaign_short",
+  "normal_target_encoding": "vision_banana_x_left_red",
+  "normal_target_resize_policy": "vector_field_renormalize_then_encode",
+  "num_train_epochs": 1,
+  "num_validation_images": 4,
+  "offload": false,
+  "optimizer": "AdamW",
+  "output_dir": "<local-path>",
+  "palette_manifold_loss_weight": 0.0,
+  "permanent_checkpoint_steps": "20000,37500",
+  "precompute_cache_dir": null,
+  "pretrained_model_name_or_path": "<local-path>",
+  "prodigy_beta3": null,
+  "prodigy_decouple": true,
+  "prodigy_safeguard_warmup": true,
+  "prodigy_use_bias_correction": true,
+  "prompt_embedding_batch_size": 16,
+  "prompt_encoding_mode": "auto",
+  "prompt_precompute_max_unique": 2048,
+  "push_to_hub": false,
+  "random_flip": false,
+  "repeats": 1,
+  "report_to": "tensorboard",
+  "resample_dataset_each_epoch": true,
+  "resolution": 512,
+  "resume_from_checkpoint": "latest",
+  "revision": null,
+  "sample_batch_size": 4,
+  "samples_per_epoch": 160000,
+  "scale_lr": false,
+  "seed": 42,
+  "skip_final_inference": true,
+  "skip_final_save": true,
+  "task_homogeneous_batches": true,
+  "train_all_transformer_blocks": true,
+  "train_batch_size": 1,
+  "train_full_transformer": false,
+  "train_input_layers": false,
+  "train_last_single_transformer_blocks": 0,
+  "train_last_transformer_blocks": 0,
+  "train_method": "partial_full_rank",
+  "train_non_block_layers": true,
+  "train_output_layers": true,
+  "train_single_transformer_block_indices": null,
+  "train_transformer_block_indices": null,
+  "unit_normal_loss_weight": 0.0,
+  "upcast_before_saving": false,
+  "use_8bit_adam": false,
+  "validation_epochs": 50,
+  "validation_height": null,
+  "validation_image": null,
+  "validation_num_inference_steps": 28,
+  "validation_prompt": null,
+  "validation_prompts_file": null,
+  "validation_width": null,
+  "variant": null,
+  "weighting_scheme": "none"
+}