HRNet / Lite-HRNet

This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network.
Apache License 2.0
819 stars 126 forks source link

about model.train() #45

Open EEWenbinWu opened 3 years ago

EEWenbinWu commented 3 years ago

Do we have to use this train func when we train a model?

def train(self, mode=True): 
       """Convert the model into training mode."""
        super().train(mode)
        if mode and self.norm_eval: 
                for m in self.modules():
                        if isinstance(m, _BatchNorm):
                            m.eval()