File size: 705 Bytes
c9abed5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
    "name": "SparseUnetVaeDecoder",
    "args": {
        "out_channels": 6,
        "model_channels": [1024, 512, 256, 128, 64],
        "latent_channels": 32,
        "num_blocks": [4, 16, 8, 4, 0],
        "block_type": [
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d"
        ],
        "up_block_type": [
            "SparseResBlockC2S3d",
            "SparseResBlockC2S3d",
            "SparseResBlockC2S3d",
            "SparseResBlockC2S3d"
        ],
        "block_args": [{}, {}, {}, {}, {}],
        "pred_subdiv": false,
        "use_fp16": true
    }
}