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