torch.optim.swa_utils.get_swa_avg_fn
-
torch.optim.swa_utils.get_swa_avg_fn()[исходный код] -
Получить функцию, применяющую стохастическое усреднение весов (SWA) к одному параметру.
© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/generated/torch.optim.swa_utils.get_swa_avg_fn.html