ddpm-lsun-cat-ema / config.json
patrickvonplaten's picture
Update config.json
0a7b8ff
{
"_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
}