Update train.py

train.py

@@ -336,6 +336,9 @@ if __name__ == "__main__":
     # here we create the datasets objects as well as tokenization and encoding
     train_data, valid_data, test_data = dataio_prepare(hparams, tokenizer)
 
+    run_on_main(hparams["pretrainer"].collect_files)
+    hparams["pretrainer"].load_collected()
+
     # Trainer initialization
     asr_brain = ASR(
         modules=hparams["modules"],