Skip to content

Commit 3f10ec0

Browse files
author
Sean Naren
authored
Merge pull request #134 from srviest/patch-2
fix bug in turning off bucketing sampler
2 parents 998904f + 2fc914a commit 3f10ec0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
parser.add_argument('--tensorboard', dest='tensorboard', action='store_true', help='Turn on tensorboard graphing')
5555
parser.add_argument('--log_dir', default='visualize/deepspeech_final', help='Location of tensorboard log')
5656
parser.add_argument('--log_params', dest='log_params', action='store_true', help='Log parameter values and gradients')
57-
parser.add_argument('--no_bucketing', dest='no_bucketing', action='store_false',
57+
parser.add_argument('--no_bucketing', dest='no_bucketing', action='store_true',
5858
help='Turn off bucketing and sample from dataset based on sequence length (smallest to largest)')
5959
parser.set_defaults(cuda=False, silent=False, checkpoint=False, visdom=False, augment=False, tensorboard=False,
6060
log_params=False, no_bucketing=False)

0 commit comments

Comments
 (0)