File size: 874 Bytes
3f93d73 55aa274 3f93d73 55aa274 3f93d73 55aa274 3f93d73 55aa274 b13e2a9 55aa274 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
{
"_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
} |