| { | |
| "algo": "diffusion_2d_cond", | |
| "b1": 0.5, | |
| "b2": 0.999, | |
| "batch_size": 32, | |
| "hf_entity": "IDEALLab", | |
| "hf_repo_prefix": "engiopt", | |
| "latent_dim": 100, | |
| "layers_per_block": 2, | |
| "lr": 0.0004, | |
| "n_cpu": 8, | |
| "n_epochs": 200, | |
| "noise_schedule": "linear", | |
| "num_timesteps": 1000, | |
| "problem_id": "photonics2d", | |
| "sample_interval": 400, | |
| "save_model": true, | |
| "seed": 2, | |
| "strict_determinism": false, | |
| "track": true, | |
| "wandb_entity": "engibench", | |
| "wandb_project": "engiopt" | |
| } |