jwieczorekhabana commited on
Commit
c84c4c7
1 Parent(s): ad3a4fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -15,8 +15,7 @@ This model only contains the `GaudiConfig` file for running the [Wav2Vec2](https
15
  This enables to specify:
16
  - `use_fused_adam`: whether to use Habana's custom AdamW implementation
17
  - `use_fused_clip_norm`: whether to use Habana's fused gradient norm clipping operator
18
- - `disable_autocast`: whether to disable autocast; this parameter takes precedence over --bf16 flag and is temporary as some scripts produce nan values.
19
- In those cases this parameter is already present in huggingface topology Habana gaudi_config.json.
20
 
21
 
22
  ## Usage
 
15
  This enables to specify:
16
  - `use_fused_adam`: whether to use Habana's custom AdamW implementation
17
  - `use_fused_clip_norm`: whether to use Habana's fused gradient norm clipping operator
18
+ - `use_torch_autocast`: whether to use autocast;
 
19
 
20
 
21
  ## Usage