Skip to content

Commit 7728a10

Browse files
committed
[src] Fix bug in online2-nnet3 decoding RE dropout+batch-norm (thanks: Wonkyum Lee)
1 parent cdb25d1 commit 7728a10

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/online2bin/online2-wav-nnet3-latgen-faster.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,8 @@ int main(int argc, char *argv[]) {
160160
Input ki(nnet3_rxfilename, &binary);
161161
trans_model.Read(ki.Stream(), binary);
162162
am_nnet.Read(ki.Stream(), binary);
163+
SetBatchnormTestMode(true, &(am_nnet.GetNnet()));
164+
SetDropoutTestMode(true, &(am_nnet.GetNnet()));
163165
}
164166

165167
// this object contains precomputed stuff that is used by all decodable

0 commit comments

Comments
 (0)