torchliter.engine.eventsEventCategoryAn enumeration.
EventHandlerBase Class for Event Handlers.
EventHandler.__init____init__(
action_function: Optional[Callable[[torchliter.engine.base.EngineBase], NoneType]] = None,
trigger_function: Optional[Callable[[torchliter.engine.base.EngineBase], bool]] = None,
**kwargs
)
EventHandler.actionaction(*args, **kwargs)
EventHandler.configconfig(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
EventHandler.triggertrigger(engine: torchliter.engine.base.EngineBase) → bool
PreEpochHandlerHanldes events when a new epoch starts.
PreEpochHandler.__init____init__(
action_function: Callable[[torchliter.engine.base.EngineBase], NoneType],
trigger_function: Optional[Callable[[torchliter.engine.base.EngineBase], bool]] = None,
every: int = 1,
train_stub: bool = True,
eval_stub: bool = True,
lambda_stub: bool = False
)
PreEpochHandler.actionaction(*args, **kwargs)
PreEpochHandler.configconfig(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PreEpochHandler.triggertrigger(engine: torchliter.engine.base.EngineBase) → bool
PostEpochHandlerHanldes events when a new epoch finishes.
PostEpochHandler.__init____init__(
action_function: Callable[[torchliter.engine.base.EngineBase], NoneType],
trigger_function: Optional[Callable[[torchliter.engine.base.EngineBase], bool]] = None,
every: int = 1,
train_stub: bool = True,
eval_stub: bool = True,
lambda_stub: bool = False
)
PostEpochHandler.actionaction(*args, **kwargs)
PostEpochHandler.configconfig(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PostEpochHandler.triggertrigger(engine: torchliter.engine.base.EngineBase) → bool
PreIterationHandlerHandles events before each iteration.
PreIterationHandler.__init____init__(
action_function: Callable[[torchliter.engine.base.EngineBase], NoneType],
trigger_function: Optional[Callable[[torchliter.engine.base.EngineBase], bool]] = None,
every: int = 1,
train_stub: bool = True,
eval_stub: bool = True,
lambda_stub: bool = False
)
PreIterationHandler.actionaction(*args, **kwargs)
PreIterationHandler.configconfig(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PreIterationHandler.triggertrigger(engine: torchliter.engine.base.EngineBase) → bool
PostIterationHandlerHandles events after each iteration.
PostIterationHandler.__init____init__(
action_function: Callable[[torchliter.engine.base.EngineBase], NoneType],
trigger_function: Optional[Callable[[torchliter.engine.base.EngineBase], bool]] = None,
every: int = 1,
train_stub: bool = True,
eval_stub: bool = True,
lambda_stub: bool = False
)
PostIterationHandler.actionaction(*args, **kwargs)
PostIterationHandler.configconfig(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PostIterationHandler.triggertrigger(engine: torchliter.engine.base.EngineBase) → bool
EngineEngine with Event Handler plugin.
Attributes ———- _event_handlers : Dict[EventCategory, List[EventHandler]] The registry of event handlers.
Engine.__init____init__()
Engine.after_iterationafter_iteration()
Engine.attach_eventattach_event(handler: torchliter.engine.events.EventHandler)
Attach an event handler.
Parameters ———- handler : EventHandler An event handler to be attached.
Engine.before_iterationbefore_iteration()
Engine.list_eventslist_events(
event_category: Optional[str] = None
) → Tuple[torchliter.engine.events.EventHandler]
List events based on category.
Parameters
———- event_category : Optional[str] List event handlers in event_category (the default is None). If not provided, list all handlers.
Returns ——- Tuple[EventHandler] Tuple of handlers.
Engine.when_epoch_finisheswhen_epoch_finishes()
Engine.when_epoch_startswhen_epoch_starts()