Method set_bn_train

pycontrast/learning/contrast_trainer.py:230–233

(m)

        model_ema.eval()

        def set_bn_train(m):
            classname = m.__class__.__name__
            if classname.find('BatchNorm') != -1:
                m.train()
        model_ema.apply(set_bn_train)

        batch_time = AverageMeter()