Add config.json

config.json

@@ -0,0 +1,176 @@
+{
+  "embedding_dim": 1728,
+  "slow_stem_channels": [
+    16,
+    32,
+    64,
+    128
+  ],
+  "slow_stem_kernels": [
+    [
+      3,
+      1
+    ],
+    [
+      3,
+      1
+    ],
+    [
+      3,
+      1
+    ],
+    [
+      3,
+      3
+    ]
+  ],
+  "slow_stem_strides": [
+    [
+      2,
+      8
+    ],
+    [
+      1,
+      1
+    ],
+    [
+      1,
+      1
+    ],
+    [
+      2,
+      2
+    ]
+  ],
+  "fast_stem_channels": [
+    2,
+    4,
+    8,
+    16
+  ],
+  "fast_stem_kernels": [
+    [
+      3,
+      3
+    ],
+    [
+      3,
+      3
+    ],
+    [
+      3,
+      3
+    ],
+    [
+      3,
+      3
+    ]
+  ],
+  "fast_stem_strides": [
+    [
+      2,
+      2
+    ],
+    [
+      1,
+      1
+    ],
+    [
+      1,
+      1
+    ],
+    [
+      2,
+      2
+    ]
+  ],
+  "slow_stage_in": [
+    256,
+    512,
+    1024,
+    4608
+  ],
+  "slow_stage_out": [
+    256,
+    512,
+    1536,
+    1536
+  ],
+  "fast_stage_in": [
+    16,
+    32,
+    64,
+    192
+  ],
+  "fast_stage_out": [
+    32,
+    64,
+    192,
+    192
+  ],
+  "slow_group_size": 128,
+  "fast_group_size": 16,
+  "block_kernels": [
+    [
+      1,
+      1
+    ],
+    [
+      1,
+      3
+    ],
+    [
+      3,
+      1
+    ],
+    [
+      1,
+      1
+    ]
+  ],
+  "fusion_input_channels": [
+    32,
+    32,
+    64,
+    192
+  ],
+  "fusion_output_channels": [
+    128,
+    256,
+    512,
+    3072
+  ],
+  "fusion_time_kernel": 7,
+  "fusion_time_stride": 4,
+  "alpha": 0.2,
+  "scaled_activation": "gelu",
+  "ws_eps": 0.0001,
+  "weight_standardization": true,
+  "sample_rate": 16000,
+  "n_fft": 2048,
+  "hop_length": 160,
+  "win_length": 400,
+  "window": "hann",
+  "n_mels": 96,
+  "fmin": 0.0,
+  "fmax": 8000.0,
+  "mel_norm": 2.0,
+  "htk": true,
+  "power": 2.0,
+  "center": true,
+  "pad_mode": "reflect",
+  "mag_compression": "log10_nonneg",
+  "slice_hop": 200,
+  "look_backward": 150,
+  "look_forward": 150,
+  "standard_normalize": true,
+  "std_floor": 0.01,
+  "provenance": {
+    "source_repo": "https://github.com/PandoraMedia/music-audio-representations",
+    "paper": "https://arxiv.org/abs/2210.03799",
+    "model": "SF-NFNet-F0 (MULE)",
+    "config": "supporting_data/configs/mule_embedding_timeline.yml",
+    "code_license": "GPL-3.0-only",
+    "weights_license": "CC-BY-NC-4.0"
+  }
+}