ModelAverager (definition) in projects: aosp_15_r20

Searched defs:ModelAverager (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/model_averaging/
H A D	averagers.py	`14 class ModelAverager(ABC): class`
/aosp_15_r20/external/pytorch/torch/distributed/optim/
H A D	post_localSGD_optimizer.py	`58 def __init__(self, optim: torch.optim.Optimizer, averager: averagers.ModelAverager):`