| 1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- # Defined in torch/csrc/monitor/python_init.cpp
- import datetime
- from enum import Enum
- from typing import Callable, Dict, List, Union
- class Aggregation(Enum):
- VALUE = ...
- MEAN = ...
- COUNT = ...
- SUM = ...
- MAX = ...
- MIN = ...
- class Stat:
- name: str
- count: int
- def __init__(
- self,
- name: str,
- aggregations: List[Aggregation],
- window_size: int,
- max_samples: int = -1,
- ) -> None: ...
- def add(self, v: float) -> None: ...
- def get(self) -> Dict[Aggregation, float]: ...
- class Event:
- name: str
- timestamp: datetime.datetime
- data: Dict[str, Union[int, float, bool, str]]
- def __init__(
- self,
- name: str,
- timestamp: datetime.datetime,
- data: Dict[str, Union[int, float, bool, str]],
- ) -> None: ...
- def log_event(e: Event) -> None: ...
- class EventHandlerHandle: ...
- def register_event_handler(handler: Callable[[Event], None]) -> EventHandlerHandle: ...
- def unregister_event_handler(handle: EventHandlerHandle) -> None: ...
|
