flame.pytorch.metrics.topk_accuracy

Module Contents

Classes

TopkAccuracy

class flame.pytorch.metrics.topk_accuracy.TopkAccuracy(name, output_transform, topk=(1,), use_jit=False)

Bases: flame.pytorch.metrics.base_metric.Metric

Parameters:

• name (Sequence[str]) –

• output_transform (Callable) –

• topk (Sequence[str]) –

• use_jit (bool) –