Fixes for Wav2Vec example (#609)

* Fixes to the example * Revert paths
Lightning-Universe · Jul 21, 2021 · 0f6bb7e · 0f6bb7e
1 parent 08b56dd
commit 0f6bb7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash_examples/speech_recognition.py b/flash_examples/speech_recognition.py
@@ -29,7 +29,7 @@
 model = SpeechRecognition(backbone="facebook/wav2vec2-base-960h")
 
 # 3. Create the trainer and finetune the model
-trainer = flash.Trainer(max_epochs=1, limit_train_batches=1, limit_test_batches=1)
+trainer = flash.Trainer(max_epochs=1)
 trainer.finetune(model, datamodule=datamodule, strategy='no_freeze')
 
 # 4. Predict on audio files!