Method __init__

train_utils.py:389–393

freeze_bn and unfreeze_bn must appear in pairs

(self)

class Bn_Controller:
    def __init__(self):
        """
        freeze_bn and unfreeze_bn must appear in pairs
        """
        self.backup = {}

    def freeze_bn(self, model):
        assert self.backup == {}