File size: 1,076 Bytes
9a3da50
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
{
  "model_type": "autoencoder",
  "sample_size": 96000,
  "sample_rate": 48000,
  "audio_channels": 1,
  "model": {
    "encoder": {
      "type": "spec_convnext",
      "config": {
        "in_channels": 1,
        "channels": 512,
        "c_mults": [1, 2, 4],
        "strides": [4, 5, 6],
        "latent_dim": 1024,
        "identity_init": false,
        "n_fft": 64,
        "hop_length": 16,
        "use_snake": true,
        "causal": true,
        "padding_mode": "reflect"
      }
    },
    "decoder": {
      "type": "oobleck",
      "config": {
        "out_channels": 1,
        "channels": 320,
        "c_mults": [1, 2, 4, 8, 16],
        "strides": [2, 4, 5, 6, 8],
        "latent_dim": 512,
        "use_snake": true,
        "final_tanh": false,
        "causal": true,
        "padding_mode": "reflect"
      }
    },
    "bottleneck": {
      "type": "vae_distillation",
      "config": {
        "latent_dim": 512,
        "distillation_proj_dim": 1024
      }
    },
    "latent_dim": 512,
    "downsampling_ratio": 1920,
    "io_channels": 1
  }
}