mvq / models / conv / config / 4.yaml
4.yaml
Raw
encoder:
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 4
  kernels: [4,3,3,3,3] 
  res_kernels: [3,1]
  act: leakyrelu
encoder2:
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 8
  kernels: [4,3,3,3,3] 
  res_kernels: [3,1]
  act: leakyrelu
decoder:
  in_channel: 0
  channel: 128
  n_res_block: 2
  n_res_channel: 64
  stride: 4
  kernels: [3,4,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: [3,4,4,4] 
  res_kernels: [3,1]
  act: relu