gaudi_config.json

{
  "use_habana_mixed_precision": true,
  "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": [
    "embedding",
    "nll_loss",
    "log_softmax",
    "cross_entropy"
  ]
}