File size: 611 Bytes
be5dd93 ea69e6f 5bbab78 ea69e6f 5bbab78 ea69e6f 5bbab78 ea69e6f be5dd93 5bbab78 ea69e6f 5bbab78 ea69e6f 5bbab78 82f0a14 5bbab78 be5dd93 |
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 |
{
"_class_name": "UNetModel",
"down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D"],
"up_blocks": ["UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
"attn_resolutions": [
16
],
"conv_resample": true,
"downsample_padding": 0,
"num_head_channels": 64,
"ch": 32,
"block_channels": [32, 64],
"ch_mult": [
1,
2
],
"resnet_eps": 1e-6,
"flip_sin_to_cos": false,
"downscale_freq_shift": 1,
"dropout": 0.0,
"in_channels": 3,
"num_res_blocks": 2,
"out_ch": 3,
"out_channels": 3,
"resamp_with_conv": true,
"resolution": 32,
"image_size": 32,
"ddpm": true
}
|