kymlcode commited on
Commit
9810f39
·
verified ·
1 Parent(s): f6b5c6f

Update configs/inference/inference_audio.yaml

Browse files
configs/inference/inference_audio.yaml CHANGED
@@ -1,13 +1,13 @@
1
  a2m_model:
2
- out_dim: 2048
3
- latent_dim: 512
4
  model_path: ./pretrained_model/wav2vec2-base-960h
5
  only_last_fetures: True
6
  from_pretrained: True
7
 
8
  a2p_model:
9
  out_dim: 6
10
- latent_dim: 512
11
  model_path: ./pretrained_model/wav2vec2-base-960h
12
  only_last_fetures: True
13
  from_pretrained: True
 
1
  a2m_model:
2
+ out_dim: 1404
3
+ latent_dim: 1024
4
  model_path: ./pretrained_model/wav2vec2-base-960h
5
  only_last_fetures: True
6
  from_pretrained: True
7
 
8
  a2p_model:
9
  out_dim: 6
10
+ latent_dim: 1024
11
  model_path: ./pretrained_model/wav2vec2-base-960h
12
  only_last_fetures: True
13
  from_pretrained: True