hugo flores garcia commited on
Commit
68c46eb
·
1 Parent(s): 507e09f
conf/generated/cat/c2f.yml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ $include:
2
+ - conf/lora/lora.yml
3
+ AudioDataset.duration: 3.0
4
+ AudioDataset.loudness_cutoff: -40.0
5
+ VampNet.embedding_dim: 1280
6
+ VampNet.n_codebooks: 14
7
+ VampNet.n_conditioning_codebooks: 4
8
+ VampNet.n_heads: 20
9
+ VampNet.n_layers: 16
10
+ fine_tune: true
11
+ fine_tune_checkpoint: ./models/vampnet/c2f.pth
12
+ save_path: ./runs/cat/c2f
13
+ train/AudioLoader.sources: &id001
14
+ - scratch/cat-audio
15
+ val/AudioLoader.sources: *id001
conf/generated/cat/coarse.yml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ $include:
2
+ - conf/lora/lora.yml
3
+ fine_tune: true
4
+ fine_tune_checkpoint: ./models/vampnet/coarse.pth
5
+ save_path: ./runs/cat/coarse
6
+ train/AudioLoader.sources: &id001
7
+ - scratch/cat-audio
8
+ val/AudioLoader.sources: *id001
conf/generated/cat/interface.yml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ AudioLoader.sources:
2
+ - - scratch/cat-audio
3
+ Interface.coarse2fine_ckpt: ./runs/cat/c2f/latest/vampnet/weights.pth
4
+ Interface.coarse_ckpt: ./runs/cat/coarse/latest/vampnet/weights.pth
5
+ Interface.codec_ckpt: ./models/vampnet/codec.pth
6
+ Interface.wavebeat_ckpt: ./models/wavebeat.pth
conf/generated/cat10/c2f.yml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ $include:
2
+ - conf/lora/lora.yml
3
+ AudioDataset.duration: 3.0
4
+ AudioDataset.loudness_cutoff: -40.0
5
+ VampNet.embedding_dim: 1280
6
+ VampNet.n_codebooks: 14
7
+ VampNet.n_conditioning_codebooks: 4
8
+ VampNet.n_heads: 20
9
+ VampNet.n_layers: 16
10
+ fine_tune: true
11
+ fine_tune_checkpoint: ./models/vampnet/c2f.pth
12
+ save_path: ./runs/cat10/c2f
13
+ train/AudioLoader.sources: &id001
14
+ - scratch/cat-audio-10s
15
+ val/AudioLoader.sources: *id001
conf/generated/cat10/coarse.yml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ $include:
2
+ - conf/lora/lora.yml
3
+ fine_tune: true
4
+ fine_tune_checkpoint: ./models/vampnet/coarse.pth
5
+ save_path: ./runs/cat10/coarse
6
+ train/AudioLoader.sources: &id001
7
+ - scratch/cat-audio-10s
8
+ val/AudioLoader.sources: *id001
conf/generated/cat10/interface.yml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ AudioLoader.sources:
2
+ - - scratch/cat-audio-10s
3
+ Interface.coarse2fine_ckpt: ./runs/cat10/c2f/latest/vampnet/weights.pth
4
+ Interface.coarse_ckpt: ./runs/cat10/coarse/latest/vampnet/weights.pth
5
+ Interface.codec_ckpt: ./models/vampnet/codec.pth
6
+ Interface.wavebeat_ckpt: ./models/wavebeat.pth