Skip to content

Commit 811bd21

Browse files
authored
[scripts] Dropout schedule fix (stop crash) (#4283)
1 parent 7b7d0b8 commit 811bd21

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

egs/wsj/s5/steps/libs/nnet3/train/dropout_schedule.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,9 @@ def get_dropout_edit_option(dropout_schedule, data_fraction, iter_):
240240
set-dropout-proportion directive works.
241241
"""
242242

243+
if data_fraction > 1.0:
244+
data_fraction = 1.0
245+
243246
if dropout_schedule is None:
244247
return ""
245248

0 commit comments

Comments
 (0)