Searched defs:ModelAverager (Results 1 – 2 of 2) sorted by relevance
14 class ModelAverager(ABC): class
58 def __init__(self, optim: torch.optim.Optimizer, averager: averagers.ModelAverager):