Update train.py

train.py

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