File size: 627 Bytes
b02b64b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
{
"hmp_bf16_ops": [
"add",
"addmm",
"bmm",
"div",
"dropout",
"gelu",
"iadd",
"linear",
"layer_norm",
"matmul",
"mm",
"rsub",
"softmax",
"truediv",
"conv2d",
"dot",
"mul",
"mean"
],
"hmp_fp32_ops": [
"embedding",
"nll_loss",
"log_softmax",
"cross_entropy",
"binary_cross_entropy"
],
"hmp_is_verbose": false,
"hmp_opt_level": "O1",
"optimum_version": "1.8.8",
"transformers_version": "4.28.1",
"use_fused_adam": true,
"use_fused_clip_norm": true,
"use_habana_mixed_precision": true,
"use_torch_autocast": false
}
|