| { | |
| "module": "zea.models.diffusion", | |
| "class_name": "DiffusionModel", | |
| "config": { | |
| "name": "diffusion_model", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "input_shape": [ | |
| 256, | |
| 256, | |
| 1 | |
| ], | |
| "input_range": [ | |
| -1, | |
| 1 | |
| ], | |
| "min_signal_rate": 0.02, | |
| "max_signal_rate": 0.95, | |
| "network_name": "unet_time_conditional", | |
| "network_kwargs": { | |
| "block_depth": 2, | |
| "widths": [32, 64, 96, 128] | |
| } | |
| }, | |
| "registered_name": "DiffusionModel", | |
| "build_config": { | |
| "input_shape": [ | |
| [ | |
| 1, | |
| 256, | |
| 256, | |
| 1 | |
| ], | |
| [ | |
| 1, | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| ] | |
| }, | |
| "compile_config": { | |
| "optimizer": { | |
| "module": "keras.optimizers", | |
| "class_name": "AdamW", | |
| "config": { | |
| "name": "adamw", | |
| "learning_rate": 9.999999747378752e-06, | |
| "weight_decay": 1e-05, | |
| "clipnorm": null, | |
| "global_clipnorm": null, | |
| "clipvalue": null, | |
| "use_ema": false, | |
| "ema_momentum": 0.99, | |
| "ema_overwrite_frequency": null, | |
| "loss_scale_factor": null, | |
| "gradient_accumulation_steps": null, | |
| "beta_1": 0.9, | |
| "beta_2": 0.999, | |
| "epsilon": 1e-07, | |
| "amsgrad": false | |
| }, | |
| "registered_name": null | |
| }, | |
| "loss": { | |
| "module": "keras.losses", | |
| "class_name": "MeanSquaredError", | |
| "config": { | |
| "name": "mean_squared_error", | |
| "reduction": "sum_over_batch_size" | |
| }, | |
| "registered_name": null | |
| }, | |
| "loss_weights": null, | |
| "metrics": null, | |
| "weighted_metrics": null, | |
| "run_eagerly": false, | |
| "steps_per_execution": 1, | |
| "jit_compile": true | |
| } | |
| } |