import torchvision.models as models MODEL = models.resnet18(pretrained=False)
How to change the input channel number with parameter?
