|
{ |
|
"_class_name": "UNetModel", |
|
"attn_resolutions": [ |
|
16 |
|
], |
|
"down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"], |
|
"up_blocks": ["UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D"], |
|
"conv_resample": true, |
|
"downsample_padding": 0, |
|
"num_head_channels": null, |
|
"ch": 128, |
|
"ch_mult": [ |
|
1, |
|
1, |
|
2, |
|
2, |
|
4, |
|
4 |
|
], |
|
"block_channels": [128, 128, 256, 256, 512, 512], |
|
"resnet_eps": 1e-6, |
|
"flip_sin_to_cos": false, |
|
"downscale_freq_shift": 1, |
|
"dropout": 0.0, |
|
"in_channels": 3, |
|
"name_or_path": "./ddpm-lsun-church/", |
|
"num_res_blocks": 2, |
|
"out_ch": 3, |
|
"out_channels": 3, |
|
"resamp_with_conv": true, |
|
"resolution": 256, |
|
"image_size": 256, |
|
"ddpm": true |
|
} |
|
|