Add training metadata and checksums: creative_fraction/editing50

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/COMPLETE.json

@@ -0,0 +1,7 @@
+{
+  "status": "COMPLETE",
+  "attempt": "3",
+  "completed_utc": "2026-08-25T19:50:30.275733+00:00",
+  "completion_source": "trainer_exit_zero",
+  "wrapper_exit_code": 0
+}

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/SHA256SUMS

@@ -0,0 +1,6 @@
+97ea39929ca88b293ce307114cfc265a8345d82675b69b3232269390cfdf09ee  config.json
+235a30e4d3ac93b6eaf483fe690af3d302d18b9e883878695fa60aff5ff0818b  diffusion_pytorch_model.safetensors.index.json
+532ea989d2205b03f474955b622e23fe4086c0e27edc35f64a5f8b414e7d3b5f  diffusion_pytorch_model-00001-of-00004.safetensors
+c6ab5d33571bbbed75f00332bfa3fa9fc15964e7eb77e0f7338547ced045fd93  diffusion_pytorch_model-00002-of-00004.safetensors
+151a4b0277e484b705b6106152ee63a82b56e0444b46c4715e8d2f672ded1b63  diffusion_pytorch_model-00003-of-00004.safetensors
+dcb37150b210fea09705fcdeac011d1ac0246d1b3f5af99f8521abca3a8a46e2  diffusion_pytorch_model-00004-of-00004.safetensors

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/checksum_validation.json

@@ -0,0 +1,30 @@
+{
+  "fingerprints": {
+    "config.json": {
+      "size": 552,
+      "mtime_ns": 1787687392000000000
+    },
+    "diffusion_pytorch_model.safetensors.index.json": {
+      "size": 24875,
+      "mtime_ns": 1787687422000000000
+    },
+    "diffusion_pytorch_model-00001-of-00004.safetensors": {
+      "size": 9804206048,
+      "mtime_ns": 1787687400000000000
+    },
+    "diffusion_pytorch_model-00002-of-00004.safetensors": {
+      "size": 9797913704,
+      "mtime_ns": 1787687408000000000
+    },
+    "diffusion_pytorch_model-00003-of-00004.safetensors": {
+      "size": 9596584288,
+      "mtime_ns": 1787687416000000000
+    },
+    "diffusion_pytorch_model-00004-of-00004.safetensors": {
+      "size": 7115649136,
+      "mtime_ns": 1787687422000000000
+    }
+  },
+  "sha256_manifest": "2bda13173addeaca1de6b25ef30e62a528cd0329299ade490c1ce0cd8c4c40a7",
+  "validation": "PASS"
+}

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/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.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "bdd"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_bdd",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "voc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.015333333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "mapillary"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_mapillary",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.015333333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "voc_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.007666666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "refseg",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.03833333333333333
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "pico_banana",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.3
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "identity",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.2
+    },
+    {
+      "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.10833333333333334
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.011666666666666667
+    }
+  ],
+  "excluded_geometry_sources": [],
+  "geometry_version": "geometry_canonical_v2_v3",
+  "planned_task_group_samples": {
+    "depth": 25333,
+    "identity": 32000,
+    "instance": 6134,
+    "pd12m": 17333,
+    "pico": 48000,
+    "refseg": 6133,
+    "semantic": 15733,
+    "surface_normals": 9334
+  },
+  "samples_per_epoch": 160000,
+  "schema_version": 2,
+  "seed": 42,
+  "shuffle": true,
+  "source_weighting": "declared_relative_weight_within_task",
+  "task_group_weights": {
+    "depth": 0.15833333333333333,
+    "identity": 0.2,
+    "instance": 0.03833333333333333,
+    "pd12m": 0.10833333333333334,
+    "pico": 0.3,
+    "refseg": 0.03833333333333333,
+    "semantic": 0.09833333333333333,
+    "surface_normals": 0.058333333333333334
+  }
+}

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/dataset_specs.resolved.json

@@ -0,0 +1,242 @@
+{
+  "datasets": [
+    {
+      "filters": {
+        "source_dataset": "acdc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_acdc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "bdd"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_bdd",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "voc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.024583333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.015333333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "mapillary"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_mapillary",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.015333333333333332
+    },
+    {
+      "filters": {
+        "source_dataset": "voc_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.007666666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "refseg",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.03833333333333333
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "pico_banana",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.3
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "identity",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.2
+    },
+    {
+      "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.10833333333333334
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.03166666666666666
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.011666666666666667
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "unique_data_fraction": 1.0,
+      "unique_data_seed": 42,
+      "weight": 0.011666666666666667
+    }
+  ],
+  "samples_per_epoch": 160000,
+  "seed": 42,
+  "shuffle": true
+}

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/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/cross_task/creative_task_fraction/editing50/checkpoint-50000/run_spec.json

@@ -0,0 +1,112 @@
+{
+  "account": "h100_v115be14_mirror",
+  "accounts_sha256": "af3ceb90fa458653c105addccc12626788c0a29171542c0c153e8b333c357792",
+  "after_run": null,
+  "campaign_file": "<local-path>",
+  "campaign_id": "geometry_phase2_20260806_v1",
+  "campaign_sha256": "8d81de74c098f1b833f3e1f1e3bcc4c9cd1aef1079f74bf682b6fa08d6f7cd75",
+  "campaign_status": "optional_phase2",
+  "created_utc": "2026-08-11T10:23:26.433224+00:00",
+  "dataset_specs_sha256": "908f840747d69b675e17f8946c7a06b606265eec8d917cdcc819109ea1a0416a",
+  "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.15833333333333333",
+    "DISTRIBUTED_STRATEGY": "fsdp",
+    "EVAL_GPU_GRES": "gpu:h100:4",
+    "EVAL_PARTITION": "h100",
+    "EVAL_QOS": "normal",
+    "EXCLUDE_DEPTH_SOURCE": "",
+    "EXCLUDE_NORMALS_SOURCE": "",
+    "FSDP_TRANSFORMER_DTYPE_POLICY": "fp32_all",
+    "GRADIENT_ACCUMULATION_STEPS": "2",
+    "GUIDANCE_SCALE": "3.5",
+    "IDENTITY_ARCHIVE": "<local-path>",
+    "IDENTITY_WEIGHT": "0.2",
+    "INSTANCE_WEIGHT": "0.03833333333333333",
+    "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.058333333333333334",
+    "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.10833333333333334",
+    "PERMANENT_CHECKPOINT_STEPS": "50000",
+    "PICO_ARCHIVE": "<local-path>",
+    "PICO_WEIGHT": "0.3",
+    "PROJECT_ROOT": "<local-path>",
+    "PROMPT_ENCODING_MODE": "auto",
+    "REFSEG_WEIGHT": "0.03833333333333333",
+    "REPORT_TO": "tensorboard",
+    "RESAMPLE_DATASET_EACH_EPOCH": "1",
+    "RESOLUTION": "512",
+    "RESUME_FROM_CHECKPOINT": "latest",
+    "ROOT": "<local-path>",
+    "RUN_DIR": "<local-path>",
+    "RUN_ID": "p2_edit_fraction_50_s42",
+    "SAMPLES_PER_EPOCH": "160000",
+    "SEED": "42",
+    "SEMANTIC_WEIGHT": "0.09833333333333333",
+    "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": "creative_fraction_sweep",
+  "full_eval_checkpoints": [
+    50000
+  ],
+  "geometry_eval_checkpoints": [
+    50000
+  ],
+  "geometry_gate": "<local-path>",
+  "geometry_gate_sha256": "762516ac340dca76c5fa3f9c45848823a32aa749582a5b2363cfaf9c6ee388eb",
+  "git_commit": "8d4f2df7242feabfe95039f2f3181894a0048c0e",
+  "group": "p2_creative_fraction_sweep",
+  "inputs_sha256": "4fab5be1229ad50bdbd21fd4406e864ae8f778f53308d7ae4df51f7e48615d4c",
+  "label": "Image editing 50% (2x default); equal-delta redistribution",
+  "run_id": "p2_edit_fraction_50_s42",
+  "schema_version": 1,
+  "status": "PLANNED"
+}

ablation_models/additional_studies/cross_task/creative_task_fraction/editing50/checkpoint-50000/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": "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"
+}