File size: 836 Bytes
7794430 b6c7533 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 343f91b 7794430 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 |
{
"use_habana_mixed_precision": true,
"hmp_is_verbose": false,
"use_fused_adam": true,
"use_fused_clip_norm": true,
"hmp_bf16_ops": [
"addmm",
"batch_norm",
"bmm",
"conv1d",
"conv2d",
"conv3d",
"conv_transpose1d",
"conv_transpose2d",
"conv_transpose3d",
"dot",
"dropout",
"dropout1d",
"dropout2d",
"dropout3d",
"group_norm",
"instance_norm",
"layer_norm",
"leaky_relu",
"linear",
"matmul",
"mean",
"mm",
"mv",
"relu",
"t"
],
"hmp_fp32_ops": [
"binary_cross_entropy",
"binary_cross_entropy_with_logits",
"cross_entropy",
"div",
"divide",
"embedding",
"embedding_bag",
"log",
"log2",
"log_softmax",
"nll_loss",
"smooth_l1_loss",
"softmax",
"topk",
"truediv"
]
}
|