Summerizer

[source]

SummaryOptions

polyaxon.models.summarizer.SummaryOptions()

[source]

SummaryTypes

polyaxon.models.summarizer.SummaryTypes()

add_learning_rate_summaries

add_learning_rate_summaries()

Adds learning rate summaries. Only works when decaying learning rate is chosen.


add_loss_summaries

add_loss_summaries(total_loss, loss)

Adds loss scalar summaries.

  • Args:

    • total_loss: Tensor. The total loss (Regression loss + regularization losses).
    • loss: Tensor. Regression loss.
  • Returns: The list of created loss summaries.


add_activations_summary

add_activations_summary(activation_ops)

Adds histogram and scalar summary for given activations.

  • Args:

    • activation_ops: A list of Tensor. The activations to summarize.
  • Returns: The list of created activation summaries.


add_gradients_summary

add_gradients_summary(grads)

Add histogram summary for given gradients and scalar summary for clipped gradients.

  • Args:

    • grads: A list of Tensor. The gradients to summarize.
  • Returns: The list of created gradient summaries.


add_trainable_vars_summary

add_trainable_vars_summary(variables)

Adds histogram summary for given variables weights.

  • Args:

    • variables: A list of Variable. The variables to summarize.
  • Returns: The list of created weights summaries.