diffusion-ecg-augmentation / model_metadata.json
TharakaDil2001's picture
Upload model_metadata.json with huggingface_hub
983c573 verified
{
"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
}