Rate this Page
★ ★ ★ ★ ★

torch.optim.swa_utils.get_swa_multi_avg_fn#

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

Get the function applying stochastic weight average (SWA) across multiple params.

On this page
Show Source
PyTorch Libraries