audio-transformers-course icon indicating copy to clipboard operation
audio-transformers-course copied to clipboard

Wrong output on chapter4/fine-tuning

Open DavidFM43 opened this issue 1 year ago • 1 comments

Hi👋, I think the output should be:

DatasetDict({
    train: Dataset({
        features: ['genre', 'input_values', 'attention_mask'],
        num_rows: 899
    })
    test: Dataset({
        features: ['genre', 'input_values', 'attention_mask'],
        num_rows: 100
    })
})

Instead of

https://github.com/huggingface/audio-transformers-course/blob/c3a8701dd376ad7c6b1e0a4a8c6970ccc2d60cb9/chapters/en/chapter4/fine-tuning.mdx#L321-L336

Since return_attention_mask=True in the feature_extractor. Is this the case?

DavidFM43 avatar Jul 14 '23 19:07 DavidFM43

@DavidFM43 you're right. Perhaps you could directly open a PR. In case you do not open a PR for this particular issue, I'll be opening one for this file and will make sure to include this change along with the other code changes.

MHRDYN7 avatar Jul 15 '23 14:07 MHRDYN7