From b9077cc609ebb6eee10b3e70e08ad5f6ad78ad18 Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Wed, 5 Jan 2022 17:19:49 +0000 Subject: [PATCH] Fixes --- flash/audio/speech_recognition/cli.py | 4 ++-- flash/audio/speech_recognition/data.py | 1 + flash_examples/speech_recognition.py | 4 ++-- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/flash/audio/speech_recognition/cli.py b/flash/audio/speech_recognition/cli.py index a74a930d25..b44470feb8 100644 --- a/flash/audio/speech_recognition/cli.py +++ b/flash/audio/speech_recognition/cli.py @@ -28,8 +28,8 @@ def from_timit( """Downloads and loads the timit data set.""" download_data("https://pl-flash-data.s3.amazonaws.com/timit_data.zip", "./data") return SpeechRecognitionData.from_json( - input_fields="file", - target_fields="text", + "file", + "text", train_file="data/timit/train.json", test_file="data/timit/test.json", val_split=val_split, diff --git a/flash/audio/speech_recognition/data.py b/flash/audio/speech_recognition/data.py index 2657aaa285..0bce404ac6 100644 --- a/flash/audio/speech_recognition/data.py +++ b/flash/audio/speech_recognition/data.py @@ -423,6 +423,7 @@ def from_datasets( objects. The Dataset objects should be one of the following: + * A PyTorch Dataset where the ``__getitem__`` returns a tuple: ``(file_path or , target)`` * A PyTorch Dataset where the ``__getitem__`` returns a dict: ``{"input": file_path, "target": target}`` diff --git a/flash_examples/speech_recognition.py b/flash_examples/speech_recognition.py index 8da2e40aeb..b3fc8eba10 100644 --- a/flash_examples/speech_recognition.py +++ b/flash_examples/speech_recognition.py @@ -21,8 +21,8 @@ download_data("https://pl-flash-data.s3.amazonaws.com/timit_data.zip", "./data") datamodule = SpeechRecognitionData.from_json( - input_fields="file", - target_fields="text", + "file", + "text", train_file="data/timit/train.json", test_file="data/timit/test.json", batch_size=4,