pytorch_tao.plugins.Metric

class pytorch_tao.plugins.Metric(name: str, metric: Metric, tune=False, direction='max')

Adapter plugin for ignite metrics. Use this plugin as a adapter for using ignite.metrics.

Parameters

• name – name of the metric score

• metric – ignite.metrics.metric.Metric instance

• tune – if tune based on this metric

import pytorch_tao as tao
from pytorch_tao.plugin import Metric
from ignite.metrics import Accuracy

trainer = tao.Trainer()
trainer.use(Metric("accuracy", Accuracy()))

__init__(name: str, metric: Metric, tune=False, direction='max')

Methods

__init__(name, metric[, tune, direction])
after_use()
attach(engine)
set_engine(engine)

Attributes