Ejemplos de generator_metric_fn_eval en Python

Ejemplo n.º 1

 def joint_metric(masked_lm_example_loss, masked_lm_log_probs,
                  masked_lm_ids, masked_lm_weights,
                  next_sentence_example_loss,
                  next_sentence_log_probs, next_sentence_labels,
                  discriminator_dict):
     generator_metric = generator_metric_fn_eval(
         masked_lm_example_loss, masked_lm_log_probs,
         masked_lm_ids, masked_lm_weights,
         next_sentence_example_loss, next_sentence_log_probs,
         next_sentence_labels)
     discriminator_metric = discriminator_metric_eval(
         discriminator_dict)
     generator_metric.update(discriminator_metric)
     return generator_metric

Ejemplo n.º 2

Archivo: electra_model_fn_fce.py Proyecto: Beleiaya/BERT

 def joint_metric(masked_lm_example_loss, masked_lm_log_probs,
                  masked_lm_ids, masked_lm_weights,
                  next_sentence_example_loss,
                  next_sentence_log_probs, next_sentence_labels,
                  per_example_loss, logits, input_ori_ids,
                  input_ids, input_mask):
     generator_metric = generator_metric_fn_eval(
         masked_lm_example_loss, masked_lm_log_probs,
         masked_lm_ids, masked_lm_weights,
         next_sentence_example_loss, next_sentence_log_probs,
         next_sentence_labels)
     discriminator_metric = discriminator_metric_eval(
         per_example_loss, logits, input_ori_ids, input_ids,
         input_mask)
     generator_metric.update(discriminator_metric)
     return generator_metric