use dynamic shapes (#2)
Browse files- use dynamic shapes (ee5bfb39d6a07235c78bc8bc2f49e77ade8eb59b)
Co-authored-by: Shiv Kaul <skaulintel@users.noreply.huggingface.co>
- gaudi_config.json +2 -1
gaudi_config.json
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"use_fused_adam": true,
|
| 3 |
"use_fused_clip_norm": true,
|
| 4 |
-
"use_torch_autocast": true
|
|
|
|
| 5 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"use_fused_adam": true,
|
| 3 |
"use_fused_clip_norm": true,
|
| 4 |
+
"use_torch_autocast": true,
|
| 5 |
+
"use_dynamic_shapes": true
|
| 6 |
}
|