Skip to content

Commit eb39efa

Browse files
committed
Fixing lossgen to work with USE_WEIGHTS_FILE
Retraining a new model so that we have a matching .pth file. Also, discarding more data on init to get better randomization.
1 parent 0e564fd commit eb39efa

File tree

3 files changed

+30
-6
lines changed

3 files changed

+30
-6
lines changed

autogen.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ set -e
99
srcdir=`dirname $0`
1010
test -n "$srcdir" && cd "$srcdir"
1111

12-
dnn/download_model.sh "8f34305a299183509d22c7ba66790f67916a0fc56028ebd4c8f7b938458f2801"
12+
dnn/download_model.sh "160753e983198f29f1aae67c54caa0e30bd90f1ce916a52f15bdad2df8e35e58"
1313

1414
echo "Updating build configuration files, please wait...."
1515

dnn/lossgen.c

+1-5
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ int sample_loss(
147147
so we skip them. */
148148
if (!st->used) {
149149
int i;
150-
for (i=0;i<100;i++) sample_loss_impl(st, percent_loss);
150+
for (i=0;i<1000;i++) sample_loss_impl(st, percent_loss);
151151
st->used = 1;
152152
}
153153
return sample_loss_impl(st, percent_loss);
@@ -157,11 +157,7 @@ void lossgen_init(LossGenState *st)
157157
{
158158
int ret;
159159
OPUS_CLEAR(st, 1);
160-
#ifndef USE_WEIGHTS_FILE
161160
ret = init_lossgen(&st->model, lossgen_arrays);
162-
#else
163-
ret = 0;
164-
#endif
165161
celt_assert(ret == 0);
166162
(void)ret;
167163
}

tar_list.txt

+28
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
dnn/models/lace_v2.pth
2+
dnn/models/nolace_v2.pth
3+
dnn/models/nolace_192_v2.pth
4+
dnn/models/nolace_160_v2.pth
5+
dnn/models/fargan48bbr_adv_50.pth
6+
dnn/models/rdovae_sparse5m_32.pth
7+
dnn/models/plc4ar_16.pth
8+
dnn/models/pitch_vsmallconv1.pth
9+
dnn/models/lossgen3_2000.pth
10+
dnn/lace_data.c
11+
dnn/lace_data.h
12+
dnn/nolace_data.c
13+
dnn/nolace_data.h
14+
dnn/fargan_data.c
15+
dnn/fargan_data.h
16+
dnn/pitchdnn_data.c
17+
dnn/pitchdnn_data.h
18+
dnn/plc_data.c
19+
dnn/plc_data.h
20+
dnn/dred_rdovae_constants.h
21+
dnn/dred_rdovae_stats_data.h
22+
dnn/dred_rdovae_enc_data.h
23+
dnn/dred_rdovae_stats_data.c
24+
dnn/dred_rdovae_dec_data.h
25+
dnn/dred_rdovae_enc_data.c
26+
dnn/dred_rdovae_dec_data.c
27+
dnn/lossgen_data.c
28+
dnn/lossgen_data.h

0 commit comments

Comments
 (0)