This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

with_cp confusing #5

Open

torinchen opened this issue Aug 16, 2022 · 1 comment

torinchen commented Aug 16, 2022

why with_cp flag is set, the norm's momentum need reset?
if with_cp:
momentum = 1 - 0.9 ** 0.5
self.norm_cfg.update(dict(momentum=momentum))

JeremieMelo commented Aug 19, 2022

https://discuss.pytorch.org/t/checkpoint-with-batchnorm-running-averages/17738

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.