{
  "models": {
    "denoiser": {
      "name": "InterleavedFlowTransformer",
      "args": {
        "resolution": 64,
        "in_channels": 64,
        "out_channels": 32,
        "model_channels": 1536,
        "cond_channels": 0,
        "num_blocks": 60,
        "v11_init_policy": "odd_copy_even_identity",
        "base_block_policy": "alternating_u_x",
        "num_heads": 12,
        "mlp_ratio": 5.3334,
        "rope_freq": [
          1,
          10000
        ],
        "dtype": "float32",
        "use_checkpoint": true,
        "checkpoint_group_size": 2,
        "share_mod": true,
        "qk_rms_norm": true,
        "partial_mean": [
          3.611770664232335,
          1.9163118034138746,
          1.952608078723815,
          0.2910785085662031,
          0.050816326157399436,
          -0.5254554342843234,
          0.30951655881595963,
          -0.8197975459866622,
          0.38620928427331275,
          -0.4964843134411725,
          -0.6825266282042826,
          1.3085809901770953,
          -1.0911941103735894,
          0.44349868701875067,
          -1.1721900675779473,
          1.1157969135230572,
          -1.135226667830614,
          2.234862037712815,
          0.28582271446763646,
          2.640323985270555,
          0.15107635964282012,
          2.165778925496729,
          -1.7239707588343525,
          -1.4536538929673588,
          0.9615435201854913,
          0.9627418033316325,
          0.17127697540739928,
          -0.044568065854129285,
          2.4527678131314383,
          0.23741388043728626,
          -0.7125081495349594,
          -0.8392259054313957
        ],
        "partial_std": [
          2.281878097994697,
          2.2069980905383275,
          2.301781262493739,
          2.2781209635645494,
          2.8149015793750904,
          1.803326133427939,
          1.7410887436600158,
          2.523892276683066,
          2.7172925946489106,
          2.1446440936869227,
          1.7059470326455453,
          2.70657116251002,
          2.386575929607397,
          1.9716654166648369,
          2.8987434709120956,
          2.412686860777434,
          1.9001049086001556,
          2.8054303614532863,
          2.938243007849802,
          1.853011888157837,
          2.35922996737155,
          2.4103167387725697,
          3.0474224154374827,
          2.386036377355109,
          2.2699966056199004,
          2.060129709179025,
          2.127143704781528,
          1.988618505693798,
          2.562611683572114,
          1.87036480863336,
          2.5588610248943686,
          2.4600089118392092
        ],
        "pbr_mean": [
          3.501659,
          2.212398,
          2.226094,
          0.251093,
          -0.026248,
          -0.687364,
          0.439898,
          -0.928075,
          0.029398,
          -0.339596,
          -0.869527,
          1.038479,
          -0.972385,
          0.126042,
          -1.129303,
          0.455149,
          -1.209521,
          2.069067,
          0.544735,
          2.569128,
          -0.323407,
          2.293,
          -1.925608,
          -1.217717,
          1.213905,
          0.971588,
          -0.023631,
          0.10675,
          2.021786,
          0.250524,
          -0.662387,
          -0.768862
        ],
        "pbr_std": [
          2.665652,
          2.743913,
          2.765121,
          2.595319,
          3.037293,
          2.291316,
          2.144656,
          2.911822,
          2.969419,
          2.501689,
          2.154811,
          3.163343,
          2.621215,
          2.381943,
          3.186697,
          3.021588,
          2.295916,
          3.234985,
          3.233086,
          2.26014,
          2.874801,
          2.810596,
          3.29272,
          2.674999,
          2.680878,
          2.372054,
          2.451546,
          2.353556,
          2.995195,
          2.379849,
          2.786195,
          2.77519
        ],
        "direct_modulation_broadcast": true,
        "safe_empty_cache": true,
        "refinement_indices": []
      }
    }
  },
  "dataset": {
    "name": "NativeObservationDataset",
    "args": {
      "resolution": 1024,
      "image_size": 1024,
      "min_aesthetic_score": 4.5,
      "max_tokens": 32768,
      "full_pbr": false,
      "bridge_z_obs": true,
      "z_obs_dir_key": "z_obs_partial",
      "bridge_z_obs_args": {
        "partial_support_mode": "partial_messages",
        "partial_z_obs_allowed_extra_keys": [
          "camera_cosine",
          "overlap_rows",
          "partner_support_dilated_hit",
          "partner_support_exact_hit",
          "partner_view",
          "primary_observed_rows",
          "primary_view",
          "secondary_added_rows",
          "secondary_observed_rows",
          "union_observed_rows"
        ]
      },
      "pbr_slat_normalization": {
        "mean": [
          3.501659,
          2.212398,
          2.226094,
          0.251093,
          -0.026248,
          -0.687364,
          0.439898,
          -0.928075,
          0.029398,
          -0.339596,
          -0.869527,
          1.038479,
          -0.972385,
          0.126042,
          -1.129303,
          0.455149,
          -1.209521,
          2.069067,
          0.544735,
          2.569128,
          -0.323407,
          2.293,
          -1.925608,
          -1.217717,
          1.213905,
          0.971588,
          -0.023631,
          0.10675,
          2.021786,
          0.250524,
          -0.662387,
          -0.768862
        ],
        "std": [
          2.665652,
          2.743913,
          2.765121,
          2.595319,
          3.037293,
          2.291316,
          2.144656,
          2.911822,
          2.969419,
          2.501689,
          2.154811,
          3.163343,
          2.621215,
          2.381943,
          3.186697,
          3.021588,
          2.295916,
          3.234985,
          3.233086,
          2.26014,
          2.874801,
          2.810596,
          3.29272,
          2.674999,
          2.680878,
          2.372054,
          2.451546,
          2.353556,
          2.995195,
          2.379849,
          2.786195,
          2.77519
        ]
      },
      "shape_slat_normalization": {
        "mean": [
          0.781296,
          0.018091,
          -0.495192,
          -0.558457,
          1.06053,
          0.093252,
          1.518149,
          -0.933218,
          -0.732996,
          2.604095,
          -0.118341,
          -2.143904,
          0.495076,
          -2.179512,
          -2.130751,
          -0.996944,
          0.261421,
          -2.217463,
          1.260067,
          -0.150213,
          3.790713,
          1.481266,
          -1.046058,
          -1.523667,
          -0.059621,
          2.22078,
          1.621212,
          0.87723,
          0.567247,
          -3.175944,
          -3.186688,
          1.578665
        ],
        "std": [
          5.972266,
          4.706852,
          5.44501,
          5.209927,
          5.32022,
          4.547237,
          5.020802,
          5.444004,
          5.226681,
          5.683095,
          4.831436,
          5.286469,
          5.652043,
          5.367606,
          5.525084,
          4.730578,
          4.805265,
          5.124013,
          5.530808,
          5.619001,
          5.10393,
          5.41767,
          5.269677,
          5.547194,
          5.634698,
          5.235274,
          6.110351,
          5.511298,
          6.237273,
          4.879207,
          5.347008,
          5.405691
        ]
      },
      "attrs": [
        "base_color",
        "metallic",
        "roughness",
        "alpha"
      ],
      "skip_aesthetic_score_datasets": [
        "texverse",
        "4d_dress",
        "human4dit",
        "thuman2.1",
        "xhumans",
        "ObjaverseXL_sketchfab_curated_aug_ge6_sft",
        "TexVerse_curated_aug_ge6_sft"
      ],
      "num_views": 16
    }
  },
  "trainer": {
    "name": "ObservationGuidedCompletionTrainer",
    "args": {
      "max_steps": 30000,
      "batch_size_per_gpu": 9,
      "batch_split": 9,
      "snapshot_batch_size": 1,
      "snapshot_num_samples": 1,
      "num_workers": 32,
      "optimizer": {
        "name": "AdamW",
        "args": {
          "lr": 0.0001,
          "weight_decay": 0.01,
          "betas": [
            0.9,
            0.95
          ],
          "eps": 1e-08
        }
      },
      "lr_scheduler": {
        "name": "ContinuationCosineLRScheduler",
        "args": {
          "source_step": 15000,
          "source_lr": 2.3369604652746925e-05,
          "total_steps": 30000,
          "min_lr": 1e-05,
          "rewarm_steps": 0,
          "rewarm_lr": 4e-05
        }
      },
      "ema_rate": [
        0.9999
      ],
      "mix_precision_mode": "amp",
      "mix_precision_dtype": "bfloat16",
      "elastic": null,
      "grad_clip": {
        "name": "AdaptiveGradClipper",
        "args": {
          "max_norm": 1,
          "clip_percentile": 95
        }
      },
      "partial_zobs_p_drop": 0.03,
      "i_print": 5,
      "i_log": 5,
      "i_sample": -1,
      "i_save": 200,
      "t_schedule": {
        "name": "logitNormal",
        "args": {
          "mean": 1.0,
          "std": 1.0
        }
      },
      "sigma_min": 1e-05,
      "resume_plan": {
        "schema": "supratex.sampler_resume.v1",
        "sampler_kind": "balanced",
        "expected_step": 17000,
        "expected_dataset_size": 571051,
        "source_topology": {
          "world_size": 96,
          "batch_size_per_gpu": 6,
          "prefetch_batches": 1
        },
        "target_topology": {
          "world_size": 64,
          "batch_size_per_gpu": 9
        },
        "expected_source_state": {
          "epoch": 17,
          "idx": 924
        },
        "expected_target_state": {
          "epoch": 17,
          "idx": 1377
        },
        "conversion_report": {
          "source_epoch": 17,
          "source_idx": 924,
          "source_prefetch_batches": 1,
          "prefetched_samples_per_rank": 6,
          "discarded_prefetch_global_samples": 576,
          "effective_source_idx": 918,
          "source_epoch_samples_per_rank": 5946,
          "saved_source_epoch_cursor": 88704,
          "effective_source_epoch_cursor": 88128,
          "target_epoch": 17,
          "target_idx": 1377,
          "target_epoch_samples_per_rank": 8919,
          "target_epoch_cursor": 88128,
          "replayed_samples": 0,
          "skipped_samples": 0
        }
      }
    }
  },
  "data_dir": "training_data.json",
  "training_data_contract": "Provide root paths and omitted dataset arguments in a local training_data.json packet; no dataset, checkpoint, or private filesystem path is bundled."
}