t5-3b-summarization-gaudi-2 / gaudi_config.json
regisss's picture
regisss HF staff
Training in progress, epoch 1
ab52cfb
raw
history blame contribute delete
No virus
506 Bytes
{
"hmp_bf16_ops": [
"add",
"addmm",
"bmm",
"div",
"dropout",
"embedding",
"gelu",
"iadd",
"layer_norm",
"linear",
"log_softmax",
"matmul",
"mm",
"nll_loss",
"rsub",
"softmax"
],
"hmp_fp32_ops": [
"cross_entropy"
],
"hmp_is_verbose": false,
"hmp_opt_level": "O1",
"optimum_version": "1.5.1",
"transformers_version": "4.24.0",
"use_fused_adam": true,
"use_fused_clip_norm": true,
"use_habana_mixed_precision": false
}