Replies: 1 comment
-
@pgsrv https://github.com/pytorch/pytorch/blob/e6779d4357ae94cc9f9fedb83a87eb6126016769/torch/optim/swa_utils.py#L115 |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi! Is there an example code of updating bn stats when running avg_checkpoints?
(discussion: #224)
Beta Was this translation helpful? Give feedback.
All reactions