{
  "config": {
    "base_models_folder": "../../data/2025-08-25-Training-3D-Inpainting/models/pconvunet_true_true_batch_size_64_lr_1e-4",
    "data": {
      "constant_values": 0,
      "dataset_csv": "../../data/2025-05-27-Training-Data-Generation/subset.csv",
      "folder": "../../data/2025-08-25-Training-3D-Inpainting",
      "mask_generation": {
        "max_masks": 3,
        "max_ratio": 0.2,
        "min_masks": 1,
        "min_ratio": 0.0
      },
      "modes": {
        "train": {
          "aug_params": {
            "gaussian_noise_sigma": [
              0.0,
              0.2
            ],
            "random_scaling_range": [
              0.7,
              1.3
            ]
          },
          "batch_size": 64,
          "num_workers": 16
        },
        "val-high-snr": {
          "batch_size": 64,
          "num_workers": 16,
          "sigma": 0.1
        },
        "val-low-snr": {
          "batch_size": 64,
          "num_workers": 16,
          "sigma": 0.3
        }
      },
      "ndim": 3,
      "noise_files": [
        "../../data/2025-05-30-3DNoise-Compilation/10_processed.npy",
        "../../data/2025-05-30-3DNoise-Compilation/11_processed.npy",
        "../../data/2025-05-30-3DNoise-Compilation/12_processed.npy",
        "../../data/2025-05-30-3DNoise-Compilation/13_processed.npy",
        "../../data/2025-05-30-3DNoise-Compilation/14_processed.npy"
      ],
      "noise_normalization": "../../data/2025-05-30-3DNoise-Compilation/normalization.npy",
      "pad_mode": "constant",
      "patch_size": 64,
      "power_threshold": 0.0001
    },
    "debug": false,
    "device": "cuda:0",
    "model": {
      "id": "pconvunet",
      "params": {
        "dim": 3,
        "layer_size": 5,
        "residual": false
      }
    },
    "seed": 42,
    "training": {
      "ema": {
        "enabled": true,
        "momentum": 0.002,
        "momentum_warmup": 0.2,
        "update_every": 5,
        "warmup_iters": 100
      },
      "epochs": 683,
      "grad_clipping_norm": 1.0,
      "loss": {
        "hole_weight": 6.0,
        "valid_weight": 1.0
      },
      "lr": 0.0001,
      "resume_from": "../../data/2025-08-25-Training-3D-Inpainting/models/pconvunet_true_true_batch_size_64_lr_1e-4/4/models/ckpt_checkpoint_560.pt",
      "step_lr_scheduler": {
        "enabled": true,
        "gamma": 0.5,
        "milestones": [
          400000,
          600000,
          700000,
          750000,
          800000
        ]
      },
      "validate_every_iteration": 1000
    }
  },
  "seed": 42
}