Function freeze_bn_layer_train

script/utils/utils.py:30–39

set batchnorm to eval() it is useful to align train and testing result

(model)

    return model

def freeze_bn_layer_train(model):
    ''' set batchnorm to eval() 
        it is useful to align train and testing result 
    '''
    # model.train()
    # print("Freezing BatchNorm Layers...")
    for module in model.modules():
        if isinstance(module, nn.BatchNorm2d):
            module.eval()
    return model

def save_image_saliancy(tensor, path, normalize: bool = False, scale_each: bool = False,):
    """