A conv layer (probably the first one) expects 4 input channels, while your data only contains 3 channels.
Change the data shape to [batch_size, 4, 224, 224] or the in_channels to 3 in the corresponding conv layer.