| { | |
| "transformers_version": "4.22.0", | |
| "model_name": "DDPM-test", | |
| "architecture": "UNet", | |
| "config": { | |
| "T": 1000, | |
| "ch": 128, | |
| "ch_mult": [1, 2, 3, 4], | |
| "attn": [2], | |
| "num_res_blocks": 2, | |
| "dropout": 0.15 | |
| }, | |
| "framework": "pytorch", | |
| "file_format": "pt", | |
| "training_config": { | |
| "epoch": 200, | |
| "batch_size": 80, | |
| "lr": 0.0001, | |
| "multiplier": 2.0, | |
| "beta_1": 0.0001, | |
| "beta_T": 0.02, | |
| "img_size": 32, | |
| "grad_clip": 1.0, | |
| "device": "cuda:0" | |
| }, | |
| "evaluation_config": { | |
| "test_load_weight": "ckpt_199_.pt", | |
| "sampled_dir": "./SampledImgs/", | |
| "sampledNoisyImgName": "NoisyNoGuidenceImgs.png", | |
| "sampledImgName": "SampledNoGuidenceImgs.png", | |
| "nrow": 8 | |
| }, | |
| "_class_name": ["diffusers.DiffusionPipeline", "DiffusionPipeline"] | |
| } |