| { |
| "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","my_pipeline"] |
| } |
