Re-initialization of the momentum

[yasserben]

Hi,

In your code, there's an argument called reset_teacher, which re-initializes the momentum network denoted as self.slow_net.
I would like to know when does reset_teacher = True, so that the momentum gets re-initialized to the state of the actual SegmentationNet denoted as self.backbone.
Because from what I understood, the MomentumNet gets initialized from the beginning to be equal to SegmentationNet, then gets updated using the exponential moving average formula and never gets re-intialized.

Thank you very much !! 😃