def attach(self, engine: "ignite.engine.Engine") -> None:
if self._name is None:
self.logger = engine.logger
if self.epoch_level:
engine.add_event_handler(Events.EPOCH_COMPLETED, self)
else:
engine.add_event_handler(Events.ITERATION_COMPLETED, self)
def attach(self, engine: "ignite.engine.Engine") -> None:
if self.epoch_level:
engine.add_event_handler(
Events.EPOCH_COMPLETED(every=self.interval), self)
else:
engine.add_event_handler(
Events.ITERATION_COMPLETED(every=self.interval), self)
def attach(self, engine: "ignite.engine.Engine") -> None:
if self._name is None:
self.logger = engine.logger
if not engine.has_event_handler(self, Events.ITERATION_COMPLETED):
engine.add_event_handler(Events.ITERATION_COMPLETED, self)
if not engine.has_event_handler(self.saver.finalize, Events.COMPLETED):
engine.add_event_handler(Events.COMPLETED,
lambda engine: self.saver.finalize())
def attach(self, engine: ignite.engine.Engine):
"""
Attach callback to ignite engine, attached method will be called on the end of each epoch
and optionally on the end of every iteration
"""
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED,
self.store)
engine.add_event_handler(ignite.engine.Events.ITERATION_COMPLETED,
self.store)
def attach(self, engine: ignite.engine.Engine):
"""Attach callback to ignite engine, attached method will be called on the end of each epoch
Args:
engine: engine that computes metrics on the end of each epoch and / or on the end of each iteration
Notes:
metrics computation plugins have to be attached before this one
"""
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED,
self.store)
engine.add_event_handler(ignite.engine.Events.ITERATION_COMPLETED,
self.store)
def attach(self, engine: "ignite.engine.Engine") -> None:
if self._name is None:
self.logger = engine.logger
if self._final_checkpoint is not None:
engine.add_event_handler(Events.COMPLETED, self.completed)
engine.add_event_handler(Events.EXCEPTION_RAISED, self.exception_raised)
if self._key_metric_checkpoint is not None:
engine.add_event_handler(Events.EPOCH_COMPLETED, self.metrics_completed)
if self._interval_checkpoint is not None:
if self.epoch_level:
engine.add_event_handler(Events.EPOCH_COMPLETED(every=self.save_interval), self.interval_completed)
else:
engine.add_event_handler(Events.ITERATION_COMPLETED(every=self.save_interval), self.interval_completed)
def attach(self, engine: "ignite.engine.Engine") -> None:
"""
Register a set of Ignite Event-Handlers to a specified Ignite engine.
Args:
engine: Ignite Engine, it can be a trainer, validator or evaluator.
"""
if not engine.has_event_handler(self.iteration_completed,
Events.ITERATION_COMPLETED):
engine.add_event_handler(Events.ITERATION_COMPLETED,
self.iteration_completed)
if not engine.has_event_handler(self.epoch_completed,
Events.EPOCH_COMPLETED):
engine.add_event_handler(Events.EPOCH_COMPLETED,
self.epoch_completed)
def attach(self, engine):
engine.add_event_handler(ignite.engine.Events.ITERATION_COMPLETED,
self.on_iteration_end)
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED,
self.on_epoch_end)
engine.add_event_handler(ignite.engine.Events.COMPLETED,
self.completed)
def attach(self, engine):
"""Attach callback to ignite engine, attached method will be called on the end of each epoch"""
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED, self.on_epoch_end)
def attach(self, engine):
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED,
self.on_epoch_end)
engine.add_event_handler(ignite.engine.Events.COMPLETED, self.store)
def attach(self, engine):
engine.add_event_handler(ignite.engine.Events.ITERATION_COMPLETED,
self.update)
def attach(self, engine):
engine.add_event_handler(ignite.engine.Events.ITERATION_COMPLETED,
self._update_from_engine)
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED,
self.reset)
def attach(self, engine: "ignite.engine.Engine"):
if self._name is None:
self.logger = engine.logger
return engine.add_event_handler(Events.STARTED, self)
def attach(self, engine):
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED, self.on_epoch_end)
engine.add_event_handler(ignite.engine.Events.COMPLETED,
lambda engine: engine.agent.save_qnet('model_final'))
def attach(self, engine):
engine.add_event_handler(ignite.engine.Events.EPOCH_COMPLETED,
self.update)
engine.add_event_handler(ignite.engine.Events.COMPLETED, self.close)
def attach(self, engine: "ignite.engine.Engine"):
return engine.add_event_handler(Events.ITERATION_COMPLETED, self)