|
{ |
|
"_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 |
|
} |
|
|