[quote=“dimple, post:1, topic:37417”]
just changed the no of channels 3 as 4 in the init class nn.Conv2d as follows:
def __init__(self, block, layers, num_classes=3, zero_init_residual=False):
    super(ResNet, self).__init__()
    self.inplanes = 64
    self.conv1 = nn.Conv2d(4, 64, kernel_size=7, stride=2, padding=3,
                           bias=False)
    self.bn1 = nn.BatchNorm2d(64)
    self.relu = nn.ReLU(inplace=True)