Solved by Pranavan_Theivendira in post #2 
              
              
                Hi, 
In your forward function you are calling conv1 layer 2 times. The second time, it should be conv2. This will fix your issue. 
Thanks