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])

compute_accumulate()

compute_average()

load_state_dict(to_load)

state_dict()

__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]