diff --git a/tools/make_lmdb.py b/tools/make_lmdb.py index 322c78c..cf891f9 100644 --- a/tools/make_lmdb.py +++ b/tools/make_lmdb.py @@ -56,4 +56,4 @@ def main(): if __name__ == '__main__': - main() \ No newline at end of file + main() diff --git a/wekws/bin/train.py b/wekws/bin/train.py index 06c7526..632a240 100644 --- a/wekws/bin/train.py +++ b/wekws/bin/train.py @@ -112,8 +112,8 @@ def main(): cv_conf['spec_aug'] = False cv_conf['shuffle'] = False - train_dataset = Dataset(args.train_data, - train_conf, + train_dataset = Dataset(args.train_data, + train_conf, reverb_lmdb=args.reverb_lmdb, noise_lmdb=args.noise_lmdb) cv_dataset = Dataset(args.cv_data, cv_conf)