It seems the posted Kaggle Kernel runs without any issues or am I missing the error message?
In any case, as you’ve already described, the batch dimension is missing, so you could check, if you are indexing the batch somewhere before passing it to the conv layer.