huggingface / huggingface/audio-transformers-course

Wrong output on chapter4/fine-tuning

Open
#95 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
MDX
Stars
521
Forks
155
Avg merge
6m
Merged PRs (30d)
1

Description

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?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.