Rate this Page
★ ★ ★ ★ ★

torch.optim.swa_utils.get_swa_avg_fn#

torch.optim.swa_utils.get_swa_avg_fn()[source]#

Get the function applying stochastic weight average (SWA) across a single param.

On this page
Show Source
PyTorch Libraries