pytorch_replace_layers.py

import torch
import torchvision

model = torchvision.models.resnet18(pretrained=True)

def bn_to_in_inplace(model):
    for name, module in model.named_children():
        if isinstance(module, torch.nn.BatchNorm2d):
            setattr(model, name, torch.nn.InstanceNorm2d(module.num_features))
        else:
            bn_to_en(module)
            
bn_to_in_inplace(model)