mvq / models / conv / config / 14.yaml
14.yaml
Raw
encoder:
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 8
  kernels: [4,3,3,3,3] 
  res_kernels: [3,1]
  act: leakyrelu
encoder2:
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 2
  kernels: [4,3]
  res_kernels: [3,1]
  act: leakyrelu
decoder:
  in_channel: 0
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 8
  kernels: [2,2,3,4,4] 
  res_kernels: [3,1]
  act: relu
decoder2:
  in_channel: 0
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 8
  kernels: [2,2,3,4,4] 
  res_kernels: [3,1]
  act: relu