File size: 516 Bytes
be5dd93 ea69e6f a04ae6d 5bbab78 5fe6538 5bbab78 ea69e6f 5fe6538 ea69e6f 5bbab78 ea69e6f 5bbab78 7c813cc 5bbab78 be5dd93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
{
"_class_name": "UNetModel",
"down_block_types": ["DownBlock2D", "AttnDownBlock2D"],
"up_block_types": ["AttnUpBlock2D", "UpBlock2D"],
"conv_resample": true,
"downsample_padding": 0,
"num_head_channels": 8,
"block_out_channels": [32, 32],
"resnet_eps": 1e-6,
"flip_sin_to_cos": false,
"downscale_freq_shift": 1,
"dropout": 0.0,
"in_channels": 3,
"num_res_blocks": 1,
"out_ch": 3,
"out_channels": 3,
"resamp_with_conv": true,
"resolution": 32,
"sample_size": 32,
"ddpm": true
}
|