primepake commited on
Commit
2202b7c
·
1 Parent(s): 201da39

update config

Browse files
Files changed (2) hide show
  1. dac-vae/configs/configx2.yml +3 -3
  2. dac-vae/run.sh +1 -0
dac-vae/configs/configx2.yml CHANGED
@@ -2,10 +2,10 @@
2
  vae:
3
  sample_rate: 24000
4
  encoder_dim: 64
5
- latent_dim: 64
6
- encoder_rates: [2, 4, 5, 8]
7
  decoder_dim: 1536
8
- decoder_rates: [8, 5, 4, 2]
9
  d_in: 1
10
  d_out: 1
11
  weight_init: xavier
 
2
  vae:
3
  sample_rate: 24000
4
  encoder_dim: 64
5
+ latent_dim: 80
6
+ encoder_rates: [2,3,4,4,5]
7
  decoder_dim: 1536
8
+ decoder_rates: [5,4,4,3,2]
9
  d_in: 1
10
  d_out: 1
11
  weight_init: xavier
dac-vae/run.sh ADDED
@@ -0,0 +1 @@
 
 
1
+ torchrun --nnodes=1 --nproc_per_node=4 train.py --run_id factorx2 --resume --tag 166k_20250827_135642