Add training metadata and checksums: shared/balanced_control_warmup1000

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/COMPLETE.json

@@ -0,0 +1,7 @@
+{
+  "status": "COMPLETE",
+  "attempt": "2",
+  "completed_utc": "2026-08-07T06:45:50.131180+00:00",
+  "completion_source": "trainer_exit_zero",
+  "wrapper_exit_code": 0
+}

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/SHA256SUMS

@@ -0,0 +1,6 @@
+be895c2602b3d07fbbe7abb8ef7322315c4d5aab5db822a4fdc5f56ee91db42a  config.json
+235a30e4d3ac93b6eaf483fe690af3d302d18b9e883878695fa60aff5ff0818b  diffusion_pytorch_model.safetensors.index.json
+a0ecdb2158eda24b27b074420323e565bce1c7b0ab7899503f74b969c46f0997  diffusion_pytorch_model-00001-of-00004.safetensors
+95337ce869a470cbbda9309df62a5dcb2c845663b018f950c2cea89e38275f5e  diffusion_pytorch_model-00002-of-00004.safetensors
+d6288b7925b864ba59c4e381c95fe7c6321151c24fe0b08aee3ec34307b4aa0d  diffusion_pytorch_model-00003-of-00004.safetensors
+24221fc7d67598ba327b28e874a40618e4a499119cfd950dfb1d866751718d0e  diffusion_pytorch_model-00004-of-00004.safetensors

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/checksum_validation.json

@@ -0,0 +1,30 @@
+{
+  "fingerprints": {
+    "config.json": {
+      "size": 563,
+      "mtime_ns": 1786006756000000000
+    },
+    "diffusion_pytorch_model.safetensors.index.json": {
+      "size": 24875,
+      "mtime_ns": 1786006785000000000
+    },
+    "diffusion_pytorch_model-00001-of-00004.safetensors": {
+      "size": 9804206048,
+      "mtime_ns": 1786006763000000000
+    },
+    "diffusion_pytorch_model-00002-of-00004.safetensors": {
+      "size": 9797913704,
+      "mtime_ns": 1786006771000000000
+    },
+    "diffusion_pytorch_model-00003-of-00004.safetensors": {
+      "size": 9596584288,
+      "mtime_ns": 1786006779000000000
+    },
+    "diffusion_pytorch_model-00004-of-00004.safetensors": {
+      "size": 7115649136,
+      "mtime_ns": 1786006785000000000
+    }
+  },
+  "sha256_manifest": "9daf69de7c7c52007605b9982fb038fdb1b24a22767cead876acca9662360ac3",
+  "validation": "PASS"
+}

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/dataset_specs.json

@@ -0,0 +1,227 @@
+{
+  "campaign_id": "geometry_phase2_20260806_v1",
+  "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.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    }
+  ],
+  "excluded_geometry_sources": [],
+  "geometry_version": "geometry_canonical_v2_v3",
+  "planned_task_group_samples": {
+    "depth": 32000,
+    "identity": 16000,
+    "instance": 12800,
+    "pd12m": 24000,
+    "pico": 24000,
+    "refseg": 12800,
+    "semantic": 22400,
+    "surface_normals": 16000
+  },
+  "samples_per_epoch": 160000,
+  "schema_version": 2,
+  "seed": 42,
+  "shuffle": true,
+  "source_weighting": "declared_relative_weight_within_task",
+  "task_group_weights": {
+    "depth": 0.2,
+    "identity": 0.1,
+    "instance": 0.08,
+    "pd12m": 0.15,
+    "pico": 0.15,
+    "refseg": 0.08,
+    "semantic": 0.14,
+    "surface_normals": 0.1
+  }
+}

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/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.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.04
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.02
+    }
+  ],
+  "samples_per_epoch": 160000,
+  "seed": 42,
+  "shuffle": true
+}

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/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/shared_controls/balanced_control_warmup1000/checkpoint-10000/run_spec.json

@@ -0,0 +1,113 @@
+{
+  "account": "h200_y506dd10",
+  "accounts_sha256": "b010afdf01023c24c9faa1753910aabe52fd190710a3cd529bf67fdece47abe0",
+  "after_run": null,
+  "campaign_file": "<local-path>",
+  "campaign_id": "geometry_phase2_20260806_v1",
+  "campaign_sha256": "4e8d71e73bad3ae0abce2511fd909ffb7453f83cd6d9ede3686d2569203e1f24",
+  "campaign_status": "optional_phase2",
+  "created_utc": "2026-08-05T23:36:57.860735+00:00",
+  "dataset_specs_sha256": "9dfb9c0f7c658e22a25ff61182278e44e35a3ac30cb7358ddba42f4b026e3b95",
+  "environment": {
+    "ALLOW_LEGACY_HYPERSIM_RANGE_DEPTH": "0",
+    "ALLOW_UNVERIFIED_NORMAL_SOURCES": "0",
+    "ANGULAR_NORMAL_AUX_WEIGHT": "0",
+    "ASPECT_RATIO_BUCKETS": "512,512;576,448;448,576;608,416;416,608",
+    "AUTO_RECONCILE_ON_COMPLETE": "0",
+    "CACHE_LATENTS": "0",
+    "CHECKPOINTING_STEPS": "1000",
+    "CHECKPOINTS_TOTAL_LIMIT": "3",
+    "CUDA_VISIBLE_DEVICES": "0,1,2,3",
+    "DATASET_POOL_SEED": "42",
+    "DATASET_SPECS": "<local-path>",
+    "DEPTH_ENCODING": "relative_inverse_grayscale",
+    "DEPTH_PROMPT_CONTRACT": "campaign_detailed_inverse",
+    "DEPTH_SOURCE_MULTIPLIERS": "{}",
+    "DEPTH_TARGET_RESIZE_POLICY": "metric_field_then_encode",
+    "DEPTH_WEIGHT": "0.2",
+    "DISTRIBUTED_STRATEGY": "fsdp",
+    "EVAL_GPU_GRES": "gpu:h200:4",
+    "EVAL_PARTITION": "h200",
+    "EVAL_QOS": "mq_y506dd",
+    "EXCLUDE_DEPTH_SOURCE": "",
+    "EXCLUDE_NORMALS_SOURCE": "",
+    "FSDP_TRANSFORMER_DTYPE_POLICY": "fp32_all",
+    "GRADIENT_ACCUMULATION_STEPS": "2",
+    "GUIDANCE_SCALE": "3.5",
+    "IDENTITY_WEIGHT": "0.10",
+    "INSTANCE_WEIGHT": "0.08",
+    "LEARNING_RATE": "2e-6",
+    "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_SCHEDULER": "constant_with_warmup",
+    "LR_WARMUP_STEPS": "1000",
+    "MAX_TRAIN_STEPS": "50000",
+    "METRIC_DEPTH_AUX_WEIGHT": "0",
+    "MIXED_PRECISION": "bf16",
+    "MODEL": "<local-path>",
+    "MODEL_FAMILY": "flux2_klein",
+    "NORMALS_WEIGHT": "0.1",
+    "NORMAL_PROMPT_CONTRACT": "campaign_short",
+    "NORMAL_SOURCE_MULTIPLIERS": "{}",
+    "NORMAL_TARGET_ENCODING": "vision_banana_x_left_red",
+    "NORMAL_TARGET_RESIZE_POLICY": "vector_field_renormalize_then_encode",
+    "NUM_PROCESSES": "4",
+    "OUTPUT_DIR": "<local-path>",
+    "PALETTE_MANIFOLD_LOSS_WEIGHT": "0",
+    "PD12M_WEIGHT": "0.15",
+    "PERMANENT_CHECKPOINT_STEPS": "5000,10000,20000,50000",
+    "PICO_WEIGHT": "0.15",
+    "PROJECT_ROOT": "<local-path>",
+    "PROMPT_ENCODING_MODE": "auto",
+    "REFSEG_WEIGHT": "0.08",
+    "REPORT_TO": "tensorboard",
+    "RESAMPLE_DATASET_EACH_EPOCH": "1",
+    "RESOLUTION": "512",
+    "RESUME_FROM_CHECKPOINT": "latest",
+    "ROOT": "<local-path>",
+    "RUN_DIR": "<local-path>",
+    "RUN_ID": "p2_control_balanced_s42",
+    "SAMPLES_PER_EPOCH": "160000",
+    "SEED": "42",
+    "SEMANTIC_WEIGHT": "0.14",
+    "SKIP_FINAL_INFERENCE": "1",
+    "SKIP_FINAL_SAVE": "1",
+    "TASK_HOMOGENEOUS_BATCHES": "1",
+    "TRAINING_SUITE": "geometry_phase2_20260806_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",
+    "UNIT_NORMAL_LOSS_WEIGHT": "0",
+    "WEIGHTING_SCHEME": "none",
+    "WEIGHT_DECAY": "1e-4"
+  },
+  "family": "combined_winner",
+  "full_eval_checkpoints": [
+    50000
+  ],
+  "geometry_eval_checkpoints": [
+    5000,
+    10000,
+    20000,
+    50000
+  ],
+  "geometry_gate": "<local-path>",
+  "geometry_gate_sha256": "762516ac340dca76c5fa3f9c45848823a32aa749582a5b2363cfaf9c6ee388eb",
+  "git_commit": "b1c7d3f04f961a80bbb3d0d856f39b7d4fd45db8",
+  "group": "p0_core",
+  "inputs_sha256": "4fab5be1229ad50bdbd21fd4406e864ae8f778f53308d7ae4df51f7e48615d4c",
+  "label": "Explicit H200 reproduction of the best-supported balanced 50K control",
+  "run_id": "p2_control_balanced_s42",
+  "schema_version": 1,
+  "status": "PLANNED"
+}

ablation_models/additional_studies/shared_controls/balanced_control_warmup1000/checkpoint-10000/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": 1000,
+  "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": 50000,
+  "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": "5000,10000,20000,50000",
+  "precompute_cache_dir": "<local-path>",
+  "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"
+}