wav2vec2 / gaudi_config.json
regisss's picture
regisss HF staff
Create Gaudi config
4f5c01e
raw
history blame
393 Bytes
{
"use_habana_mixed_precision": true,
"hmp_opt_level": "O1",
"hmp_is_verbose": false,
"use_fused_adam": true,
"use_fused_clip_norm": true,
"hmp_bf16_ops": [
"addmm",
"iadd",
"linear",
"matmul",
"mm",
"mv",
"conv1d",
"conv2d",
"instance_norm"
],
"hmp_fp32_ops": [
"cross_entropy",
"embedding",
"nll_loss",
"log_softmax"
]
}