assets/model/training_config.json

{
  "data_dir": "/project/luofeng/feiyang/MaterialGeneration/hybrid_diffusion_material_generation_quarter_version_5_cubic/processed_data/",
  "output_dir": "/project/luofeng/feiyang/MaterialGeneration/hybrid_diffusion_material_generation_quarter_version_5_cubic/checkpoints/config_1_quarter_v5_cubic/",
  "experiment_dir": "/project/luofeng/feiyang/MaterialGeneration/hybrid_diffusion_material_generation_quarter_version_5_cubic/checkpoints/config_1_quarter_v5_cubic/exp_20260430_134155",
  "timestamp": "20260430_134155",
  "batch_size": 32,
  "num_epochs": 100,
  "learning_rate": 0.0001,
  "T": 100,
  "n_max_layer": 5,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "lambda_angle": 1.0,
  "lambda_material": 1.0,
  "lambda_vf_category": 1.0,
  "lambda_layer": 1.0,
  "save_every": 5,
  "normalization_method": "zscore",
  "random_seed": 533,
  "device": "cuda:0",
  "gen_eval_n_samples": 100,
  "gen_eval_every": 5,
  "use_discrete_angles": false,
  "save_visualizations": true,
  "warmup_ratio": 0.01,
  "log_every_n_steps": 100,
  "angle_resolution": 1.0,
  "model_config": {
    "n_conditions": 21,
    "n_materials": 4,
    "n_vf_categories": 5,
    "n_max_layer": 5,
    "d_model": 256,
    "n_heads": 4,
    "n_layers": 6,
    "dropout": 0.0
  },
  "mask_ids": {
    "material": 4,
    "vf_category": 5,
    "layer": 2
  },
  "n_relationships": 7
}