cerebras_pytorch.metrics#

Metrics#

A collection of evaluation metrics that can be used to evaluate the performance of a trained model on the Cerebras Wafer Scale Cluster.

Available metrics include:

AccuracyMetric

PerplexityMetric

DiceCoefficientMetric

MeanIOUMetric

FBetaScoreMetric

metrics.Metric#

metrics.AccuracyMetric#

metrics.PerplexityMetric#

metrics.DiceCoefficientMetric#

metrics.MeanIOUMetric#

metrics.FBetaScoreMetric#

metrics.compute_all_metrics#

previous

cerebras_pytorch.sparse

next

Writing a Custom Training Loop