{
  "model_name": "Enhanced Diffusion Counterfactual Generator",
  "total_parameters": 19111683,
  "content_encoder_params": 4273408,
  "style_encoder_params": 566786,
  "unet_params": 14271489,
  "training_time_hours": 21.635153891245523,
  "total_epochs": 100,
  "stage1_epochs": 50,
  "stage2_epochs": 50,
  "final_recon_correlation": 0.015070401688741493,
  "final_flip_n2a": 0.20000000298023224,
  "final_flip_a2n": 0.800000011920929
}