I am a beginner in deep learning, especially in designing neural networks from scratch and I am trying to build a B/W image colorizer using the Autoencoder concept in Pytorc