RuntimeError: Given groups=1, weight of size 64 1 3 3, expected input[1, 4, 224, 224] to have 1 channels, but got 4 channels instead

[ssalome (Salome)]

the image shape is [1, 224, 224]