torchliter.engine.events
EventCategory
An enumeration.
EventHandler
Base 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.action
action(*args, **kwargs)
EventHandler.config
config(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
EventHandler.trigger
trigger(engine: torchliter.engine.base.EngineBase) → bool
PreEpochHandler
Hanldes 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.action
action(*args, **kwargs)
PreEpochHandler.config
config(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PreEpochHandler.trigger
trigger(engine: torchliter.engine.base.EngineBase) → bool
PostEpochHandler
Hanldes 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.action
action(*args, **kwargs)
PostEpochHandler.config
config(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PostEpochHandler.trigger
trigger(engine: torchliter.engine.base.EngineBase) → bool
PreIterationHandler
Handles 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.action
action(*args, **kwargs)
PreIterationHandler.config
config(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PreIterationHandler.trigger
trigger(engine: torchliter.engine.base.EngineBase) → bool
PostIterationHandler
Handles 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.action
action(*args, **kwargs)
PostIterationHandler.config
config(*args, **kwargs) → Type
Add additional config kwargs.
For example ====================
@EventHandler.config(param1=1.0, param2=2.0) def some_action_function(engine): …
PostIterationHandler.trigger
trigger(engine: torchliter.engine.base.EngineBase) → bool
Engine
Engine with Event Handler plugin.
Attributes ———- _event_handlers : Dict[EventCategory, List[EventHandler]] The registry of event handlers.
Engine.__init__
__init__()
Engine.after_iteration
after_iteration()
Engine.attach_event
attach_event(handler: torchliter.engine.events.EventHandler)
Attach an event handler.
Parameters ———- handler : EventHandler An event handler to be attached.
Engine.before_iteration
before_iteration()
Engine.list_events
list_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_finishes
when_epoch_finishes()
Engine.when_epoch_starts
when_epoch_starts()