File size: 401 Bytes
8d7f03b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
input_channels: 1280
output_channels: 1280
encode_channels: 1280
decode_channels: 1280
code_dim: 1280
codebook_num: 1
codebook_size: 1024
bias: true
enc_ratios: [ 1, 1 ]
dec_ratios: [ 1, 1 ]
enc_strides: [ 1, 1 ] # no downsampling
dec_strides: [ 1, 1 ]
enc_kernel_size: 3
dec_kernel_size: 3
enc_block_dilations: [ 1, 1 ]
enc_block_kernel_size: 3
dec_block_dilations: [ 1, 1 ]
dec_block_kernel_size: 3 |