Update train.py
Browse files
train.py
CHANGED
@@ -297,8 +297,8 @@ if __name__ == "__main__":
|
|
297 |
# here we create the datasets objects as well as tokenization and encoding
|
298 |
(train_data, valid_data, test_data) = dataio_prepare(hparams)
|
299 |
|
300 |
-
run_on_main(hparams["pretrainer"].collect_files)
|
301 |
-
hparams["pretrainer"].load_collected()
|
302 |
|
303 |
# Trainer initialization
|
304 |
asr_brain = ASR(
|
|
|
297 |
# here we create the datasets objects as well as tokenization and encoding
|
298 |
(train_data, valid_data, test_data) = dataio_prepare(hparams)
|
299 |
|
300 |
+
# run_on_main(hparams["pretrainer"].collect_files)
|
301 |
+
# hparams["pretrainer"].load_collected()
|
302 |
|
303 |
# Trainer initialization
|
304 |
asr_brain = ASR(
|