cifar10-ncsnpp-ve / config.json
patrickvonplaten's picture
add model cifar
ca1c8ae
raw
history blame contribute delete
804 Bytes
{
"_class_name": "NCSNpp",
"_diffusers_version": "0.0.4",
"attention_type": "ddpm",
"attn_resolutions": [
16
],
"centered": false,
"ch_mult": [
1,
2,
2,
2
],
"conditional": true,
"continuous": true,
"conv_size": 3,
"dropout": 0.0,
"embedding_type": "fourier",
"fir": true,
"fir_kernel": [
1,
3,
3,
1
],
"fourier_scale": 16,
"image_size": 32,
"init_scale": 0.0,
"name_or_path": "/home/patrick/cifar10-ncsnpp-ve",
"nf": 128,
"nonlinearity": "swish",
"normalization": "GroupNorm",
"num_channels": 3,
"num_res_blocks": 4,
"progressive": "none",
"progressive_combine": "sum",
"progressive_input": "residual",
"resamp_with_conv": true,
"resblock_type": "biggan",
"scale_by_sigma": true,
"skip_rescale": true
}