File size: 761 Bytes
638304a 8682899 8137e62 68141d1 638304a 8137e62 638304a 5987f53 638304a 8137e62 638304a |
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 |
{
"_class_name": "UNetModel",
"attn_resolutions": [
16
],
"down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D"],
"up_blocks": ["UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
"conv_resample": true,
"downsample_padding": 0,
"num_head_channels": null,
"ch": 128,
"ch_mult": [
1,
2,
2,
2
],
"block_channels": [128, 256, 256, 256],
"resnet_eps": 1e-6,
"flip_sin_to_cos": false,
"downscale_freq_shift": 1,
"dropout": 0.1,
"in_channels": 3,
"out_channels": 3,
"name_or_path": "./ddpm-cifar10/",
"num_res_blocks": 2,
"out_ch": 3,
"resamp_with_conv": true,
"resolution": 32,
"image_size": 32,
"ddpm": true
}
|