Hugo Flores Garcia commited on
Commit
b688797
1 Parent(s): 183d21c

update confs

Browse files
Files changed (2) hide show
  1. conf/vampnet-c2f.yml +4 -0
  2. conf/vampnet.yml +1 -1
conf/vampnet-c2f.yml CHANGED
@@ -4,6 +4,10 @@ $include:
4
  VampNet.n_codebooks: 9
5
  VampNet.n_conditioning_codebooks: 3
6
 
 
 
 
 
7
  train/AudioDataset.duration: 3
8
  val/AudioDataset.duration: 3
9
  test/AudioDataset.duration: 3
 
4
  VampNet.n_codebooks: 9
5
  VampNet.n_conditioning_codebooks: 3
6
 
7
+ VampNet.embedding_dim: 1280
8
+ VampNet.n_layers: 16
9
+ VampNet.n_heads: 20
10
+
11
  train/AudioDataset.duration: 3
12
  val/AudioDataset.duration: 3
13
  test/AudioDataset.duration: 3
conf/vampnet.yml CHANGED
@@ -1,7 +1,7 @@
1
 
2
  codec_ckpt: /u/home/src/runs/codec-ckpt/codec.pth
3
  save_path: ckpt
4
- max_epochs: 1000000
5
  epoch_length: 1000
6
  save_audio_epochs: 2
7
  val_idx: [0,1,2,3,4,5,6,7,8,9]
 
1
 
2
  codec_ckpt: /u/home/src/runs/codec-ckpt/codec.pth
3
  save_path: ckpt
4
+ max_epochs: 1000
5
  epoch_length: 1000
6
  save_audio_epochs: 2
7
  val_idx: [0,1,2,3,4,5,6,7,8,9]