| { | |
| "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 | |
| } |