pytorch_pfn_extras.training.extensions.accumulate.average_accumulate.AverageSummary#
- class pytorch_pfn_extras.training.extensions.accumulate.average_accumulate.AverageSummary#
Bases:
SummaryBase
Methods
__init__
()add
(value[, weight])load_state_dict
(to_load)- __init__()#
- Return type:
None
- add(value, weight=1)#
- Parameters:
value (Union[Tensor, ndarray, floating, float, Callable[[], float]]) –
weight (Union[Tensor, ndarray, floating, float]) –
- Return type:
None
- compute_accumulate()#
- Return type:
Union[Tensor, ndarray, floating, float]
- compute_average()#
- Return type:
Union[Tensor, ndarray, floating, float]
- load_state_dict(to_load)#
- Parameters:
to_load (Dict[str, Any]) –
- Return type:
None
- state_dict()#
- Return type:
Dict[str, Any]