ITO-Master / networks /configs.yaml
jhtonyKoo's picture
first push
2e66664
raw
history blame
771 Bytes
# model architecture configurations
# Music Effects Encoder
Effects_Encoder:
default:
channels: [16, 32, 64, 128, 256, 256, 512, 512, 1024, 1024, 2048, 2048]
kernels: [25, 25, 15, 15, 10, 10, 10, 10, 5, 5, 5, 5]
strides: [4, 4, 2, 2, 2, 2, 2, 2, 2, 2, 1, 1]
dilation: [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
bias: True
norm: 'batch'
conv_block: 'res'
activation: "relu"
# TCN
TCN:
default:
condition_dimension: 2048
nblocks: 14
dilation_growth: 2
kernel_size: 15
stride: 1
channel_width: 128
stack_size: 15
causal: False
param_mapping:
condition_dimension: 2048
nblocks: 14
dilation_growth: 2
kernel_size: 15
stride: 2
channel_width: 128
stack_size: 15
causal: False