Work in local Jupyter lab but shows error in Google Colab: "… variable shape (1, 1, 256, 512) and value shape (512, 128, 1, 1) are incompatible

Question

I have a Python program which implements CNN based classification for three species of weed seeds tested on my local Jupyter Notebook (installed using Anaconda) and I saved