{
  "architecture": "StyleUNet",
  "generator": {
    "embed_dim": 256,
    "in_channels": 1,
    "out_channels": 1,
    "base_filters": 32,
    "use_FiLM": true
  },
  "discriminator": {
    "type": "NoisePatchGAN",
    "meta_embed_dim": 128,
    "base_filters": 64
  },
  "cimp_encoder": {
    "hf_repo": "Stemson-AI/cmmp-resnet18-512",
    "image_encoder": "resnet18",
    "embed_dim": 128,
    "meta_hidden_dim": 256,
    "meta_num_layers": 3,
    "crop_size": 512
  },
  "training": {
    "crop_size": 512,
    "batch_size": 16,
    "epochs": 250,
    "lr_G": 0.0002,
    "lr_D": 0.0002,
    "lambda_GAN": 1.0,
    "lambda_cycle": 0.125,
    "lambda_emb": 0.5,
    "lambda_id": 0.125,
    "id_loss_fn": "lpips",
    "cycle_loss_fn": "lpips",
    "optimizer": "Adam",
    "adversarial_loss": "LSGAN"
  },
  "meta_dim": 7,
  "meta_names": [
    "pixel_size",
    "dwell_time",
    "convergence_angle",
    "beam_current",
    "gain",
    "offset",
    "inner_coll_angle"
  ],
  "best_epoch_composite_val": 216
}