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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331
|
/* Auto generated from checkpoint nolace_small.pth (sha1: 953bf5854e1a33e8892da48a29b19aff3a272902) */
#ifndef NOLACE_DATA_H
#define NOLACE_DATA_H
#include "nnet.h"
#define NOLACE_PREEMPH 0.85f
#define NOLACE_FRAME_SIZE 80
#define NOLACE_OVERLAP_SIZE 40
#define NOLACE_NUM_FEATURES 93
#define NOLACE_PITCH_MAX 300
#define NOLACE_PITCH_EMBEDDING_DIM 64
#define NOLACE_NUMBITS_RANGE_LOW 50
#define NOLACE_NUMBITS_RANGE_HIGH 650
#define NOLACE_NUMBITS_EMBEDDING_DIM 8
#define NOLACE_COND_DIM 160
#define NOLACE_HIDDEN_FEATURE_DIM 96
#define NOLACE_NUMBITS_SCALE_0 1.0357311964035034f
#define NOLACE_NUMBITS_SCALE_1 1.735559105873108f
#define NOLACE_NUMBITS_SCALE_2 3.6004557609558105f
#define NOLACE_NUMBITS_SCALE_3 4.552478313446045f
#define NOLACE_NUMBITS_SCALE_4 5.932559490203857f
#define NOLACE_NUMBITS_SCALE_5 7.176970481872559f
#define NOLACE_NUMBITS_SCALE_6 8.114998817443848f
#define NOLACE_NUMBITS_SCALE_7 8.77063274383545f
#define NOLACE_PITCH_EMBEDDING_OUT_SIZE 64
#define NOLACE_FNET_CONV1_OUT_SIZE 96
#define NOLACE_FNET_CONV1_IN_SIZE 173
#define NOLACE_FNET_CONV1_STATE_SIZE (173 * (0))
#define NOLACE_FNET_CONV1_DELAY 0
#define NOLACE_FNET_CONV2_OUT_SIZE 160
#define NOLACE_FNET_CONV2_IN_SIZE 384
#define NOLACE_FNET_CONV2_STATE_SIZE (384 * (1))
#define NOLACE_FNET_CONV2_DELAY 0
#define NOLACE_FNET_TCONV_KERNEL_SIZE 4
#define NOLACE_FNET_TCONV_STRIDE 4
#define NOLACE_FNET_TCONV_IN_CHANNELS 160
#define NOLACE_FNET_TCONV_OUT_CHANNELS 160
#define NOLACE_FNET_GRU_OUT_SIZE 160
#define NOLACE_FNET_GRU_STATE_SIZE 160
#define NOLACE_CF1_FILTER_GAIN_A 0.690776f
#define NOLACE_CF1_FILTER_GAIN_B 0.000000f
#define NOLACE_CF1_LOG_GAIN_LIMIT 1.151293f
#define NOLACE_CF1_KERNEL_SIZE 16
#define NOLACE_CF1_LEFT_PADDING 8
#define NOLACE_CF1_FRAME_SIZE 80
#define NOLACE_CF1_OVERLAP_SIZE 40
#define NOLACE_CF1_IN_CHANNELS 1
#define NOLACE_CF1_OUT_CHANNELS 1
#define NOLACE_CF1_NORM_P 2
#define NOLACE_CF1_FEATURE_DIM 160
#define NOLACE_CF1_MAX_LAG 301
#define NOLACE_CF1_KERNEL_OUT_SIZE 16
#define NOLACE_CF1_GAIN_OUT_SIZE 1
#define NOLACE_CF1_GLOBAL_GAIN_OUT_SIZE 1
#define NOLACE_CF2_FILTER_GAIN_A 0.690776f
#define NOLACE_CF2_FILTER_GAIN_B 0.000000f
#define NOLACE_CF2_LOG_GAIN_LIMIT 1.151293f
#define NOLACE_CF2_KERNEL_SIZE 16
#define NOLACE_CF2_LEFT_PADDING 8
#define NOLACE_CF2_FRAME_SIZE 80
#define NOLACE_CF2_OVERLAP_SIZE 40
#define NOLACE_CF2_IN_CHANNELS 1
#define NOLACE_CF2_OUT_CHANNELS 1
#define NOLACE_CF2_NORM_P 2
#define NOLACE_CF2_FEATURE_DIM 160
#define NOLACE_CF2_MAX_LAG 301
#define NOLACE_CF2_KERNEL_OUT_SIZE 16
#define NOLACE_CF2_GAIN_OUT_SIZE 1
#define NOLACE_CF2_GLOBAL_GAIN_OUT_SIZE 1
#define NOLACE_AF1_FILTER_GAIN_A 1.381551f
#define NOLACE_AF1_FILTER_GAIN_B 0.000000f
#define NOLACE_AF1_SHAPE_GAIN 1.000000f
#define NOLACE_AF1_KERNEL_SIZE 16
#define NOLACE_AF1_FRAME_SIZE 80
#define NOLACE_AF1_LEFT_PADDING 15
#define NOLACE_AF1_OVERLAP_SIZE 40
#define NOLACE_AF1_IN_CHANNELS 1
#define NOLACE_AF1_OUT_CHANNELS 2
#define NOLACE_AF1_NORM_P 2
#define NOLACE_AF1_FEATURE_DIM 160
#define NOLACE_AF1_KERNEL_OUT_SIZE 32
#define NOLACE_AF1_GAIN_OUT_SIZE 2
#define NOLACE_TDSHAPE1_FEATURE_DIM 160
#define NOLACE_TDSHAPE1_FRAME_SIZE 80
#define NOLACE_TDSHAPE1_AVG_POOL_K 4
#define NOLACE_TDSHAPE1_INNOVATE 0
#define NOLACE_TDSHAPE1_POOL_AFTER 0
#define NOLACE_TDSHAPE1_ALPHA1_F_OUT_SIZE 80
#define NOLACE_TDSHAPE1_ALPHA1_F_IN_SIZE 160
#define NOLACE_TDSHAPE1_ALPHA1_F_STATE_SIZE (160 * (1))
#define NOLACE_TDSHAPE1_ALPHA1_F_DELAY 0
#define NOLACE_TDSHAPE1_ALPHA1_T_OUT_SIZE 80
#define NOLACE_TDSHAPE1_ALPHA1_T_IN_SIZE 21
#define NOLACE_TDSHAPE1_ALPHA1_T_STATE_SIZE (21 * (1))
#define NOLACE_TDSHAPE1_ALPHA1_T_DELAY 0
#define NOLACE_TDSHAPE1_ALPHA2_OUT_SIZE 80
#define NOLACE_TDSHAPE1_ALPHA2_IN_SIZE 80
#define NOLACE_TDSHAPE1_ALPHA2_STATE_SIZE (80 * (1))
#define NOLACE_TDSHAPE1_ALPHA2_DELAY 0
#define NOLACE_TDSHAPE2_FEATURE_DIM 160
#define NOLACE_TDSHAPE2_FRAME_SIZE 80
#define NOLACE_TDSHAPE2_AVG_POOL_K 4
#define NOLACE_TDSHAPE2_INNOVATE 0
#define NOLACE_TDSHAPE2_POOL_AFTER 0
#define NOLACE_TDSHAPE2_ALPHA1_F_OUT_SIZE 80
#define NOLACE_TDSHAPE2_ALPHA1_F_IN_SIZE 160
#define NOLACE_TDSHAPE2_ALPHA1_F_STATE_SIZE (160 * (1))
#define NOLACE_TDSHAPE2_ALPHA1_F_DELAY 0
#define NOLACE_TDSHAPE2_ALPHA1_T_OUT_SIZE 80
#define NOLACE_TDSHAPE2_ALPHA1_T_IN_SIZE 21
#define NOLACE_TDSHAPE2_ALPHA1_T_STATE_SIZE (21 * (1))
#define NOLACE_TDSHAPE2_ALPHA1_T_DELAY 0
#define NOLACE_TDSHAPE2_ALPHA2_OUT_SIZE 80
#define NOLACE_TDSHAPE2_ALPHA2_IN_SIZE 80
#define NOLACE_TDSHAPE2_ALPHA2_STATE_SIZE (80 * (1))
#define NOLACE_TDSHAPE2_ALPHA2_DELAY 0
#define NOLACE_TDSHAPE3_FEATURE_DIM 160
#define NOLACE_TDSHAPE3_FRAME_SIZE 80
#define NOLACE_TDSHAPE3_AVG_POOL_K 4
#define NOLACE_TDSHAPE3_INNOVATE 0
#define NOLACE_TDSHAPE3_POOL_AFTER 0
#define NOLACE_TDSHAPE3_ALPHA1_F_OUT_SIZE 80
#define NOLACE_TDSHAPE3_ALPHA1_F_IN_SIZE 160
#define NOLACE_TDSHAPE3_ALPHA1_F_STATE_SIZE (160 * (1))
#define NOLACE_TDSHAPE3_ALPHA1_F_DELAY 0
#define NOLACE_TDSHAPE3_ALPHA1_T_OUT_SIZE 80
#define NOLACE_TDSHAPE3_ALPHA1_T_IN_SIZE 21
#define NOLACE_TDSHAPE3_ALPHA1_T_STATE_SIZE (21 * (1))
#define NOLACE_TDSHAPE3_ALPHA1_T_DELAY 0
#define NOLACE_TDSHAPE3_ALPHA2_OUT_SIZE 80
#define NOLACE_TDSHAPE3_ALPHA2_IN_SIZE 80
#define NOLACE_TDSHAPE3_ALPHA2_STATE_SIZE (80 * (1))
#define NOLACE_TDSHAPE3_ALPHA2_DELAY 0
#define NOLACE_AF2_FILTER_GAIN_A 1.381551f
#define NOLACE_AF2_FILTER_GAIN_B 0.000000f
#define NOLACE_AF2_SHAPE_GAIN 1.000000f
#define NOLACE_AF2_KERNEL_SIZE 16
#define NOLACE_AF2_FRAME_SIZE 80
#define NOLACE_AF2_LEFT_PADDING 15
#define NOLACE_AF2_OVERLAP_SIZE 40
#define NOLACE_AF2_IN_CHANNELS 2
#define NOLACE_AF2_OUT_CHANNELS 2
#define NOLACE_AF2_NORM_P 2
#define NOLACE_AF2_FEATURE_DIM 160
#define NOLACE_AF2_KERNEL_OUT_SIZE 64
#define NOLACE_AF2_GAIN_OUT_SIZE 2
#define NOLACE_AF3_FILTER_GAIN_A 1.381551f
#define NOLACE_AF3_FILTER_GAIN_B 0.000000f
#define NOLACE_AF3_SHAPE_GAIN 1.000000f
#define NOLACE_AF3_KERNEL_SIZE 16
#define NOLACE_AF3_FRAME_SIZE 80
#define NOLACE_AF3_LEFT_PADDING 15
#define NOLACE_AF3_OVERLAP_SIZE 40
#define NOLACE_AF3_IN_CHANNELS 2
#define NOLACE_AF3_OUT_CHANNELS 2
#define NOLACE_AF3_NORM_P 2
#define NOLACE_AF3_FEATURE_DIM 160
#define NOLACE_AF3_KERNEL_OUT_SIZE 64
#define NOLACE_AF3_GAIN_OUT_SIZE 2
#define NOLACE_AF4_FILTER_GAIN_A 1.381551f
#define NOLACE_AF4_FILTER_GAIN_B 0.000000f
#define NOLACE_AF4_SHAPE_GAIN 1.000000f
#define NOLACE_AF4_KERNEL_SIZE 16
#define NOLACE_AF4_FRAME_SIZE 80
#define NOLACE_AF4_LEFT_PADDING 15
#define NOLACE_AF4_OVERLAP_SIZE 40
#define NOLACE_AF4_IN_CHANNELS 2
#define NOLACE_AF4_OUT_CHANNELS 1
#define NOLACE_AF4_NORM_P 2
#define NOLACE_AF4_FEATURE_DIM 160
#define NOLACE_AF4_KERNEL_OUT_SIZE 32
#define NOLACE_AF4_GAIN_OUT_SIZE 1
#define NOLACE_POST_CF1_OUT_SIZE 160
#define NOLACE_POST_CF1_IN_SIZE 160
#define NOLACE_POST_CF1_STATE_SIZE (160 * (1))
#define NOLACE_POST_CF1_DELAY 0
#define NOLACE_POST_CF2_OUT_SIZE 160
#define NOLACE_POST_CF2_IN_SIZE 160
#define NOLACE_POST_CF2_STATE_SIZE (160 * (1))
#define NOLACE_POST_CF2_DELAY 0
#define NOLACE_POST_AF1_OUT_SIZE 160
#define NOLACE_POST_AF1_IN_SIZE 160
#define NOLACE_POST_AF1_STATE_SIZE (160 * (1))
#define NOLACE_POST_AF1_DELAY 0
#define NOLACE_POST_AF2_OUT_SIZE 160
#define NOLACE_POST_AF2_IN_SIZE 160
#define NOLACE_POST_AF2_STATE_SIZE (160 * (1))
#define NOLACE_POST_AF2_DELAY 0
#define NOLACE_POST_AF3_OUT_SIZE 160
#define NOLACE_POST_AF3_IN_SIZE 160
#define NOLACE_POST_AF3_STATE_SIZE (160 * (1))
#define NOLACE_POST_AF3_DELAY 0
typedef struct {
LinearLayer nolace_pitch_embedding;
LinearLayer nolace_fnet_conv1;
LinearLayer nolace_fnet_conv2;
LinearLayer nolace_fnet_tconv;
LinearLayer nolace_fnet_gru_input;
LinearLayer nolace_fnet_gru_recurrent;
LinearLayer nolace_cf1_kernel;
LinearLayer nolace_cf1_gain;
LinearLayer nolace_cf1_global_gain;
LinearLayer nolace_cf2_kernel;
LinearLayer nolace_cf2_gain;
LinearLayer nolace_cf2_global_gain;
LinearLayer nolace_af1_kernel;
LinearLayer nolace_af1_gain;
LinearLayer nolace_tdshape1_alpha1_f;
LinearLayer nolace_tdshape1_alpha1_t;
LinearLayer nolace_tdshape1_alpha2;
LinearLayer nolace_tdshape2_alpha1_f;
LinearLayer nolace_tdshape2_alpha1_t;
LinearLayer nolace_tdshape2_alpha2;
LinearLayer nolace_tdshape3_alpha1_f;
LinearLayer nolace_tdshape3_alpha1_t;
LinearLayer nolace_tdshape3_alpha2;
LinearLayer nolace_af2_kernel;
LinearLayer nolace_af2_gain;
LinearLayer nolace_af3_kernel;
LinearLayer nolace_af3_gain;
LinearLayer nolace_af4_kernel;
LinearLayer nolace_af4_gain;
LinearLayer nolace_post_cf1;
LinearLayer nolace_post_cf2;
LinearLayer nolace_post_af1;
LinearLayer nolace_post_af2;
LinearLayer nolace_post_af3;
} NOLACELayers;
int init_nolacelayers(NOLACELayers *model, const WeightArray *arrays);
#endif /* NOLACE_DATA_H */
|