@Muhammad_Izaz    have a look at this tutorial, I also followed it,

  
      
      Medium – 17 Dec 20
  
  
    

Train a CNN using Skorch for MNIST digit recognition 17

A library that combines the flexibility of scikit-learn and the power of PyTorch

  Reading time: 5 min read