Commit
•
799cb20
1
Parent(s):
e609a60
Update README.md
Browse files
README.md
CHANGED
|
@@ -17,11 +17,11 @@ This enables to specify:
|
|
| 17 |
- `use_fused_clip_norm`: whether to use Habana's fused gradient norm clipping operator
|
| 18 |
- `use_torch_autocast`: whether to use Torch Autocast for managing mixed precision
|
| 19 |
|
| 20 |
-
|
| 21 |
## Usage
|
| 22 |
|
| 23 |
The model is instantiated the same way as in the Transformers library.
|
| 24 |
-
The only difference is that there are a few new training arguments specific to HPUs
|
|
|
|
| 25 |
|
| 26 |
[Here](https://github.com/huggingface/optimum-habana/blob/main/examples/audio-classification/run_audio_classification.py) is an audio classification example script to fine-tune a model. You can run it with Wav2Vec2 with the following command:
|
| 27 |
```bash
|
|
|
|
| 17 |
- `use_fused_clip_norm`: whether to use Habana's fused gradient norm clipping operator
|
| 18 |
- `use_torch_autocast`: whether to use Torch Autocast for managing mixed precision
|
| 19 |
|
|
|
|
| 20 |
## Usage
|
| 21 |
|
| 22 |
The model is instantiated the same way as in the Transformers library.
|
| 23 |
+
The only difference is that there are a few new training arguments specific to HPUs.\
|
| 24 |
+
This model is supported only in mixed precision training with bf16 type.
|
| 25 |
|
| 26 |
[Here](https://github.com/huggingface/optimum-habana/blob/main/examples/audio-classification/run_audio_classification.py) is an audio classification example script to fine-tune a model. You can run it with Wav2Vec2 with the following command:
|
| 27 |
```bash
|