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
|
/* Auto generated from checkpoint lace_v2.pth (sha1: 41eaab33c6cbdb192d14f43c9f292856cab789e9) */
#ifndef LACE_DATA_H
#define LACE_DATA_H
#include "nnet.h"
#define LACE_PREEMPH 0.85f
#define LACE_FRAME_SIZE 80
#define LACE_OVERLAP_SIZE 40
#define LACE_NUM_FEATURES 93
#define LACE_PITCH_MAX 300
#define LACE_PITCH_EMBEDDING_DIM 64
#define LACE_NUMBITS_RANGE_LOW 50
#define LACE_NUMBITS_RANGE_HIGH 650
#define LACE_NUMBITS_EMBEDDING_DIM 8
#define LACE_COND_DIM 128
#define LACE_HIDDEN_FEATURE_DIM 96
#define LACE_NUMBITS_SCALE_0 1.0983514785766602f
#define LACE_NUMBITS_SCALE_1 2.0509142875671387f
#define LACE_NUMBITS_SCALE_2 3.5729939937591553f
#define LACE_NUMBITS_SCALE_3 4.478035926818848f
#define LACE_NUMBITS_SCALE_4 5.926519393920898f
#define LACE_NUMBITS_SCALE_5 7.152282238006592f
#define LACE_NUMBITS_SCALE_6 8.277412414550781f
#define LACE_NUMBITS_SCALE_7 8.926830291748047f
#define LACE_PITCH_EMBEDDING_OUT_SIZE 64
#define LACE_FNET_CONV1_OUT_SIZE 96
#define LACE_FNET_CONV1_IN_SIZE 173
#define LACE_FNET_CONV1_STATE_SIZE (173 * (0))
#define LACE_FNET_CONV1_DELAY 0
#define LACE_FNET_CONV2_OUT_SIZE 128
#define LACE_FNET_CONV2_IN_SIZE 384
#define LACE_FNET_CONV2_STATE_SIZE (384 * (1))
#define LACE_FNET_CONV2_DELAY 0
#define LACE_FNET_TCONV_KERNEL_SIZE 4
#define LACE_FNET_TCONV_STRIDE 4
#define LACE_FNET_TCONV_IN_CHANNELS 128
#define LACE_FNET_TCONV_OUT_CHANNELS 128
#define LACE_FNET_GRU_OUT_SIZE 128
#define LACE_FNET_GRU_STATE_SIZE 128
#define LACE_CF1_FILTER_GAIN_A 0.690776f
#define LACE_CF1_FILTER_GAIN_B 0.000000f
#define LACE_CF1_LOG_GAIN_LIMIT 1.151293f
#define LACE_CF1_KERNEL_SIZE 16
#define LACE_CF1_LEFT_PADDING 8
#define LACE_CF1_FRAME_SIZE 80
#define LACE_CF1_OVERLAP_SIZE 40
#define LACE_CF1_IN_CHANNELS 1
#define LACE_CF1_OUT_CHANNELS 1
#define LACE_CF1_NORM_P 2
#define LACE_CF1_FEATURE_DIM 128
#define LACE_CF1_MAX_LAG 301
#define LACE_CF1_KERNEL_OUT_SIZE 16
#define LACE_CF1_GAIN_OUT_SIZE 1
#define LACE_CF1_GLOBAL_GAIN_OUT_SIZE 1
#define LACE_CF2_FILTER_GAIN_A 0.690776f
#define LACE_CF2_FILTER_GAIN_B 0.000000f
#define LACE_CF2_LOG_GAIN_LIMIT 1.151293f
#define LACE_CF2_KERNEL_SIZE 16
#define LACE_CF2_LEFT_PADDING 8
#define LACE_CF2_FRAME_SIZE 80
#define LACE_CF2_OVERLAP_SIZE 40
#define LACE_CF2_IN_CHANNELS 1
#define LACE_CF2_OUT_CHANNELS 1
#define LACE_CF2_NORM_P 2
#define LACE_CF2_FEATURE_DIM 128
#define LACE_CF2_MAX_LAG 301
#define LACE_CF2_KERNEL_OUT_SIZE 16
#define LACE_CF2_GAIN_OUT_SIZE 1
#define LACE_CF2_GLOBAL_GAIN_OUT_SIZE 1
#define LACE_AF1_FILTER_GAIN_A 1.381551f
#define LACE_AF1_FILTER_GAIN_B 0.000000f
#define LACE_AF1_SHAPE_GAIN 1.000000f
#define LACE_AF1_KERNEL_SIZE 16
#define LACE_AF1_FRAME_SIZE 80
#define LACE_AF1_LEFT_PADDING 15
#define LACE_AF1_OVERLAP_SIZE 40
#define LACE_AF1_IN_CHANNELS 1
#define LACE_AF1_OUT_CHANNELS 1
#define LACE_AF1_NORM_P 2
#define LACE_AF1_FEATURE_DIM 128
#define LACE_AF1_KERNEL_OUT_SIZE 16
#define LACE_AF1_GAIN_OUT_SIZE 1
typedef struct {
LinearLayer lace_pitch_embedding;
LinearLayer lace_fnet_conv1;
LinearLayer lace_fnet_conv2;
LinearLayer lace_fnet_tconv;
LinearLayer lace_fnet_gru_input;
LinearLayer lace_fnet_gru_recurrent;
LinearLayer lace_cf1_kernel;
LinearLayer lace_cf1_gain;
LinearLayer lace_cf1_global_gain;
LinearLayer lace_cf2_kernel;
LinearLayer lace_cf2_gain;
LinearLayer lace_cf2_global_gain;
LinearLayer lace_af1_kernel;
LinearLayer lace_af1_gain;
} LACELayers;
int init_lacelayers(LACELayers *model, const WeightArray *arrays);
#endif /* LACE_DATA_H */
|
